Pytorch: Mass resolved matrix

Considering:

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

      

Goal 1:

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

      

Goal 2:

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

      

Result 1:

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

      

Result 2:

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

      

My PyTorch version is 0.1.12_2 - would be greatly appreciated for possible solutions.

+3


source to share





All Articles