What is the difference between mm and spmm in Pytorch

user_10796158 0
Viewed 1690

What is the difference between mm and spmm in Pytorch? I know that spmm does sparse matrix multiplication, but what exactly does that mean? Why would you ever choose mm as opposed to spmm if spmm has the potential to save space?

pytorch matrix-multiplication
1 Answers

In order to use spmm you need your tensor arguments to actually be of sparse type.

Although torch.sparse representation does have the potential of saving space, sparse support does not yet covers all tensor operations and functions.

user_17144100
Related