pytorch - 如何在pytorch中按元素将稠密矩阵乘以稀疏矩阵

Question

我可以用 torch.sparse.mm()或 torch.spmm()直接在稀疏矩阵和密集矩阵之间进行乘法，但是我应该选择哪个函数进行元素乘法？

Answer 1

你可以自己实现这个乘法

def sparse_dense_mul(s, d):
  i = s._indices()
  v = s._values()
  dv = d[i[0,:], i[1,:]]  # get values from relevant entries of dense matrix
  return torch.sparse.FloatTensor(i, v * dv, s.size())

请注意，由于乘法运算的线性，您无需担心 s是否合并。