How to implement the functionality of SCATTER_MEAN function from torch_scatter

I want to replicate teh effect of Scatter_mean function in the torch-scatter library in basic pytorch code, how can I go about this?

You can use tensor.scatter_reduce_ with reduce="mean".

1 Like