fail to mask adj using 'mul'

I have got an adjacency name ‘adj’ which is a SparseTensor, I would like to mask the value in ‘adj’ using ‘mask’, which is a torch.Tensor, I try this:

from torch_sparse import  mul
...
new_adj = mul(adj, mask)

but it throw ’ ValueError: Size mismatch: Expected size (251, 1, …) or (1, 251, …), but got size torch.Size([251, 251]).’ How could I mask an adjacency matrix when it is a SparseTensor.

[rusty1s]

You can use mul_nnz as well 😃

[aries-M]

This issue had no activity for 6 months. It will be closed in 2 weeks unless there is some new activity. Is this issue already resolved?

Yes, this issue is resolved.