Pytorch scatter_理解

scatter_(dim, index, src)將src中數據根據index中的索引按照dim的方向填充進調用scatter_的向量中； 1.dim=0 首先了解，dim=0，表示按行填充，列數不變；scatter_參數中給定的index tensor，兩個中括號中數值表示x(2,5)中第一行和第二行下標的數列數不變分別被填充到向量(3,5)的第一行和第二行；數值的下標表示取x中對應下標的值，然

>>阅读原文<<