torch.gather

gather的作用：根据维度(dim)和索引(index)获取输入(input)的某些特殊值。

根据官方文档中的例子(Example)，t是2×2张量：索引是2×2张张量：

按dim=1，也就是列维度取索引值(2×2索引就是0或1代表第一列和第二列)，所以根据索引与t的对应关系得到gather后的结果：

解释：根据对应的位置和索引值确定

t的第1列元素并对应索引位置，对应的第1行，是4，第0列元素是3，按照索引的对应位置得到gather后的结果。