torch.index_reduce

torch.index_reduce(input, dim, index, source, reduce, *, include_self=True, out=None) → 张量

参见 index_reduce_() 函数描述。