torch.matmul和python内置@运算符进行矩阵乘法的区别

Question

我可以用 python 内置的 @ 运算符替换 torch.matmul 来进行矩阵乘法吗？请假设我知道 torch.matmul、torch.mm 和许多其他之间的区别。我只是想确保其中有多少可以被 @ 运算符安全地替换，而不牺牲速度或 torch 的一些本机支持。

如果没有什么坏处，我想在未来广泛使用它们。

Answer 1

是的，您始终可以使用@代替.matmul. 您也可以在 中执行相同的操作numpy。另请参阅PyTorch 将运算符映射到函数