标签： pytorch tensor
我想像这样索引一个张量：
A = [[4,5]，[6,7]]，索引= [[0,2]，[2,1]]-（？）-> [[nan，nan，6]，[ 4，7，5]]。
基本上，索引中的元素是A中对应元素所属的输出数组中的列，而行索引只是根据索引中列号的出现次数进行堆叠。
由于numpy具有更复杂的数组索引，我可以将其转换为numpy，进行转换并转换回我的ML管道中间的张量吗？我可以使用pytorch做到吗？