柳杰的博客

torch.Tensor.scatter_(dim, index, src, reduce=None) 理解scatter操作:tensor_A.scatter_(dim, index, tensor_B): tensor_B的每个元素，都按照 index 被scatter（可以理解为填充）到目标tensor_A中。(1) index和源tensor_B维度一致;(2) tensor_A一般是全零的张量，其某些特定位置的值由 tensor_B 中的值填充。(3) 注意如何根据index选取ten