pytorch中的gather方法

gather的本质是构建一个索引矩阵来获取数据

gather需要两个参数（dim, index），其中index为一个索引列表，里面元素为要修改为的索引。其形状和最终生成的结果形状一致，且与要修改矩阵的位置一一对应（如上图）。dim表示要修改哪一个维度的索引，例如，对于二维矩阵来说，dim=0则修改索引第一维，dim=1则修改第二维（如上图）