Pytorch scatter_理解

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_