torch.gather

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的含义。