526互联

torch.scatter

发布时间 2023-04-28 16:19:00作者: 放学别跑啊

(22条消息) torch.scatter算子详解_leo0308的博客-CSDN博客

image-20230428160221410

了解这个之前需要首先了解dim=0和dim=1的区别

dim=0

image-20230428160337634

dim=1

image-20230428160421465

我们看a

首先a，b是一个3×5的全零矩阵

对a执行scatter操作（将src中的值，按照指定的位置更新到input当中）

image-20230428160708529

0.5984放到第0个位置，0.1656放到第1个位置，0.8471放到第2个位置，0.3930放到第0个位置，0.2162放到第0个位置（这是在dim=0的情况下）

image-20230428160658448

dim=1时依次类推