介绍
对于稀疏矩阵,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, valueB, m, k

本文介绍了在Win10系统中安装PyTorch Sparse模块时遇到的问题及解决方案。由于Windows Python需要Visual C++库,故需要下载并安装Visual Studio Community 2017,确保包含C++ build tools和特定版本的VC++工具集。安装完成后,才能成功安装pytorch_sparse。
最低0.47元/天 解锁文章
1330





