torch.scatter

torch.scatter(input, dim, index, src) → Tensor

独立版本的 torch.Tensor.scatter_()