python - Pytorch:稀疏矩阵乘法

鉴于:

self.A = torch.autograd.Variable(random_sparse(n = dim))
self.w = torch.autograd.Variable(torch.Tensor(np.random.normal(0,1,(dim,dim))))

目标 1:

torch.mm(self.A, self.w)

目标 2:

torch.spmm(self.A, self.w)

结果1:

TypeError: Type torch.sparse.FloatTensor doesn't implement stateless method addmm

结果2:

AttributeError: 'module' object has no attribute 'spmm'

我的 PyTorch 版本是 0.1.12_2 - 非常感谢可能的解决方案。

最佳答案

spmm 已从 torch 模块移至 torch.sparse 模块。有关官方文档，请查看此 link 。 torch.sparse 模块的文档开头也有警告:

关于python - Pytorch:稀疏矩阵乘法，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/44417500/