torch.Tensor.index_reduce

Tensor.index_reduce()