Pytorch中的mm和spmm有什么区别

Question

Pytorch中的mm和spmm有什么区别？我知道 spmm 执行稀疏矩阵乘法，但这到底是什么意思？如果 spmm 有可能节省 space，你为什么会选择 mm 而不是 spmm？

Answer 1

为了使用spmm you need your tensor arguments to actually be of sparse类型。

尽管 torch.sparse 表示确实具有节省 space 的潜力，但稀疏支持尚未涵盖所有张量运算和函数。