qq_40994908的博客

Pytorch的使用：torch.gather函数**torch.gather()**作用：方便从批量tensor中获取特定化维度指定索引下的数据，该索引往往是乱序的。首先看一下官方文档中的3维数据index 代表输入向量dim 代表替换的维度input 代表最终选取的元素接下来我们用一个二维的数据，分别采用官方文档的思路和我个人理解的思路进行简化举例我们采用了一个3×3的二维矩阵进行练习输出结果1.index为行向量，dim = 0 替换行索引计算思路1：计算思路2：输出结果2.index为列向量，dim