Hi everyone,
I am using a torch.sparse_coo_tensor and I was hoping to do mean reduction along a dimension where the zeroed values are ignored.
The sparse library supports summation and even softmax, but no way to do mean.
Is there a simple trick to getting the mean to work with a sparse tensor using the index values?
Thanks,
Matt