Sparse matrix - vector multiplication

Hi,
I would like to implement a multiplication between a sparse matrix and dense vector, the result should be a dense vector.
I guess it is implemented but I couldn’t find it.
Thanks a lot!

I think

torch.mm(sparse, dense.unsqueeze(1))

should work, if dense is a dense 1D tensor.

1 Like

Yes it worked :slight_smile: thanks