介绍
对于稀疏矩阵,PyTorch (version : 1.1.0) 中提供了稀疏张量 torch.sparse.FloatTensor 来表示,其中,torch.sparse.mm(mat1, mat2) 用来完成两矩阵相乘,参数如下:
- mat1 (SparseTensor) :稀疏矩阵,大小为(n × m)
- mat2 (Tensor) :密集矩阵,大小为 (m × p)
详情参考 PyTorch 官方文档
但是嘞,我操作的矩阵太大了,将其中一个变为密集矩阵都不可实现,于是,想寻找可以直接将两个稀疏矩阵进行乘法运算的工具,然后,就找到了 pytorch_sparse 其中就有两稀疏矩阵相乘的方法:
torch_sparse.spspmm(indexA, valueA, indexB