Tensor.scatter_add_(dim, index, src) → Tensor
out.scatter_add_(dim, index, src)
1.参数:
-
dim (int) – 哪一dim进行操作
-
index (LongTensor) – 要在的out的哪一index进行操作
-
src (Tensor) – 待操作的源数字
2.官方的解释的操作如下:
3.例子
- 关于scatter_add函数的用法-CSDN博客
- 这个是自己写的俩个例子;左右个一个;最后的结果:每一行放的是与该目标节点相连的源节点的特征向量的和,并没有目标节点自身的信息,就是只聚合了邻居的消息放在这儿;
- src表示 源节点 的特征向量的集合;index表示的是 edge_index 第2行,也就是目标节点的索引,然后经过广播得到现在看到的inde;out初始值:是[ |V|,节点特征维度值]大小,初始值均为0的tensor;