pytorch中的scatter和 scatter_add 操作

Tensor的scatter和scatter_add操作，本文内容较为深入，在此进行简要梳理。以a.\text{scatter}(dim=\text{dim}, index=\text{index}, src=\text{src})为例，在具体实现过程中可参考以下代码片段：其中具体实现方式可通过以下代码片段体现：其中a = \text{torch.tensor}([1, 2, 3])；维度设置为\text{dim}；索引位置通过\text{index}指定；源数据存入变量\text{src}。

表示 a 、 index 和 src 在各维度上的规模均相等。\n\n 在除第 dim 维之外的所有维度上（ i\neq dim ）， a \text{ 的尺寸至少不小于 } index 对应维度的尺寸。\n\n 至于第 dim 维本身，则二者之间并无明确的大于或小于关系。\n\n 其中对于每个位置 j\in [0,\ size-1] （假设从0开始计数），其对应的取值范围为 [0,\ a_j - 1] 。

3. 在索引与源之间的关系中：其中，在第i个维度上的长度满足index.size(i)\leq src.size(i)（即source在每