gpt4 book ai didi

pytorch - Pytorch中的mm和spmm有什么区别

转载 作者:行者123 更新时间:2023-12-05 05:02:34 27 4
gpt4 key购买 nike

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

最佳答案

为了使用spmm你需要你的张量参数实际上是 sparse类型。

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

关于pytorch - Pytorch中的mm和spmm有什么区别,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/62125442/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com