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

Question

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

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

Answer 1

是的，您始终可以使用@ 代替.matmul。您也可以在numpy 中执行相同的操作。另见PyTorch mapping operators to functions