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时依次类推