torch.scatter
(22条消息) torch.scatter算子详解_leo0308的博客-CSDN博客
了解这个之前需要首先了解dim=0和dim=1的区别
dim=0
dim=1
我们看a
首先a,b是一个3×5的全零矩阵
对a执行scatter操作(将src中的值,按照指定的位置更新到input当中)
0.5984放到第0个位置,0.1656放到第1个位置,0.8471放到第2个位置,0.3930放到第0个位置,0.2162放到第0个位置(这是在dim=0的情况下)
dim=1时依次类推