pytorch, numpy利用另一个数组的索引进行重排序,如torch.gather.......
构造不同维度上的索引进行重组直接构造不同维度索引的方法在pytorch和numpy是类似的,具体如下:1、numpyx=np.random.randint(0,10,[2,3])print(x)结果:[[151][964]]indices=np.argsort(x,dim=1)#在第1维度上进行排序print(indices)结果:array([[0,2,1],[2,1,0]])first_ind