Pytorch scatter_理解

scatter_(dim, index, src)将src中数据根据index中的索引按照dim的方向填充进调用scatter_的向量中; 1.dim=0 首先了解,dim=0,表示按行填充,列数不变;scatter_参数中给定的index tensor,两个中括号中数值表示x(2,5)中第一行和第二行下标的数列数不变分别被填充到向量(3,5)的第一行和第二行;数值的下标表示取x中对应下标的值,然
相关文章
相关标签/搜索