解释下torch中的scatter_add_

scatter_add_(dim, index, src) -> Tensor

将张量 src 中的所有值根据 index 张量指定的位置加到 self 中，类似于 ~torch.Tensor.scatter_ 的操作。对于 src 中的每个值：

• 如果维度 dimension != dim，索引由其在 src 中的索引决定；
• 如果维度 dimension = dim，索引由 index 中的对应值决定。

对于一个三维张量，self 的更新规则如下：

self