PyTorch 的所有运算符是什么,它们的功能等价物是什么?
例如,a @ b等价于a.mm(b)或a.matmul(b)?
a @ b
a.mm(b)
a.matmul(b)
我在寻找运算符 -> 函数映射的规范列表。
我很高兴收到 PyTorch 文档链接作为答案 - 我的 googlefu 无法找到它。
operator-overloading pytorch
operator-overloading ×1
pytorch ×1