pytorch中的gather方法 2022-11-042022-09-06 by hunt gather的本质是构建一个索引矩阵来获取数据 gather需要两个参数(dim, index),其中index为一个索引列表,里面元素为要修改为的索引。其形状和最终生成的结果形状一致,且与要修改矩阵的位置一一对应(如上图)。dim表示要修改哪一个维度的索引,例如,对于二维矩阵来说,dim=0则修改索引第一维,dim=1则修改第二维(如上图)