Difference between this function and torch.Tensor.scatter_reduce_

[cinntamani]

Hello,

It seems that this function achieves the same effect as torch.Tensor.scatter_reduce_(dim, index, src, reduce, *, include_self=True) in native PyTorch (https://pytorch.org/docs/stable/generated/torch.Tensor.scatter_reduce_.html).

May I ask what are the pros and cons of using either version?

Thank you!

[SamGalanakis]

I see "The backward pass is implemented only for src.shape == index.shape." on the native torch implementation and I believe he backward pass is implemented here in all cases.