kZjPBD.jpg
2022年4月22日
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 阅读全文
posted @ 2022-04-22 20:06 Through_The_Night 阅读(33) 评论(0) 推荐(0) 编辑