torch.Tensor.select_scatter

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

参见 torch.select_scatter()