torch.scatter

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

Out-of-place version of torch.Tensor.scatter_()