qq_43430381的博客

self [index[i][j]] [j] += src[i] [j] # if dim == 0 self [i] [index[i][j]] += src[i] [j] # if dim == 1 理解scatter_add()函数，看src就行了，src有多少个，self坐标就会变多少次。 self是一个二维的数组，self[第一维，第二维]，dim==0，就是将src对应坐标，对应到 index 坐标里面的值，放置到self的第一维中。 例如src[i,