【教程】win10 使用 python3 安装 pytorch_sparse

介绍

对于稀疏矩阵,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
  • 6
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值