scatter_(dim, index, src)将src中数据根据index中的索引按照dim的方向填充进调用scatter_的向量中;
1.dim=0
首先了解,dim=0,表示按行填充,列数不变;scatter_参数中给定的index tensor,idx两个中括号对应x,比如x的size是2x5,那么indx矩阵也应该是2x5,然后因为dim=0,按行填充,行变列不变,所以比如矩阵中0,1,2索引都表示把x的值,填充到新矩阵的0,1,2行;
具体解释:比如第一个中括号第一个值0,表示取x中第0行第0列(0.3992),填充到(3,5)中的第0行第0列;第二个中括号(第二行)第一个值2,表示取x中第1行第0个值(0.5735),填充到(3,5)中的第2行第0列;
2.dim=1
dim=1,表示按列填充,行数不变;同理上面例子,因为dim=1,按列填充,列变行不变,所以比如矩阵中0,1,2索引都表示把x的值,填充到新矩阵的0,1,2列,新来的值覆写旧的值;
;链接:https://pytorch.org/docs/stable/tensors.html?highlight=scatter_#torch.Tensor.scatter_