关于torch.gather函数
直接上代码
example1
输出为
example2
输出为
可以看出,gather的作用是这样的,index实际上是索引,具体是行还是列的索引要看前面dim 的指定,[1,2;3,4],指定dim=1,也就是横向,那么索引就是列号。index的大小就是输出的大小,所以比如index是[0,0;1,0],那么看index第一行指的就是第一列,两个都为0即都指的是第一列第一个元素。同理,第二行为[1,0] ,即是第二列索引,为[4,3]这样就输入为[1,1;4,3],即可理解gather的含义。