kZjPBD.jpg

torch.scatter

https://zhuanlan.zhihu.com/p/426906301

 

 self[index[i][j][k]][j][k] = src[i][j][k]  # if dim == 0
 self[i][index[i][j][k]][k] = src[i][j][k]  # if dim == 1
 self[i][j][index[i][j][k]] = src[i][j][k]  # if dim == 2

posted @ 2022-04-22 20:06  Through_The_Night  阅读(31)  评论(0编辑  收藏  举报