PyTorch.gather()理解

函数调用 使用torch.gather()的方式有两种,第一种为直接使用torch.gather().第二种为SourceTensor.gather().容易出现疑惑的地方为dim的取值.dim的取值范围为{-2,0,1}.下面分情况进行讨论(在二维平面上讨论). dim=0 通过上图中的代码执行结果可以看出,当dim为0时,y中的数据值表示行索引,而y中的数据索引表示列索引.具体到上图中的情况即

>>阅读原文<<