torch.Tensor.select_scatter

Tensor.select_scatter(src, dim, index) → Tensor

See torch.select_scatter()