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

Question

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

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

标签： pythonmachine-learningdeep-learningpytorchmatrix-multiplication

Answer 1

是的，您始终可以使用@. .matmul您也可以在numpy. 另请参阅PyTorch 将运算符映射到函数