Pytorch:稀疏矩阵乘法

时间:2017-06-07 15:53:57

标签: python sparse-matrix pytorch

鉴于:

int? i = row["CommentPk"].ToNullableInt();


public static int? ToNullableInt(this object value)
{
    if (value == null || value == DBNull.Value)
        return null;

    return Convert.ToInt32(value);
}

目标1:

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

目标2:

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

结果1:

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

结果2:

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

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

1 个答案:

答案 0 :(得分:1)

spmm已从torch模块移至torch.sparse模块。有关官方文档,请检查此linktorch.sparse模块的文档开头也有一个警告:

  

该API目前处于实验阶段,可能会在不久的将来进行更改。