Pytorch中的mm和spmm之间有什么区别

Question

Pytorch中的mm和spmm有什么区别？我知道spmm会进行稀疏矩阵乘法，但这到底是什么意思？如果spmm有节省空间的潜力，为什么您会选择mm而不是spmm？

Answer 1

为了使用spmm，您需要张量参数实际上是spmm类型。

尽管sparse表示确实有节省空间的潜力，但稀疏支持尚未涵盖所有张量操作和函数。