gpt4 book ai didi

PyTorch 将运算符映射到函数

转载 作者:行者123 更新时间:2023-12-02 09:10:08 28 4
gpt4 key购买 nike

PyTorch 的所有运算符是什么,它们的功能等价物是什么?

例如,是 a @ b相当于 a.mm(b)a.matmul(b) ?

我在寻找运算符 -> 函数映射的规范列表。

我很高兴收到 PyTorch 文档链接作为答案 - 我的 googlefu 无法找到它。

最佳答案

Python 文档表 Mapping Operators to Functions提供来自以下方面的规范映射:

运算符 -> __function__()
例如:

Matrix Multiplication        a @ b        matmul(a, b)


在页面的其他地方,您将看到 __matmul__作为 matmul 的替代名称.

PyTorch 的定义 __functions__可以在:
  • torch.Tensor module documentation
  • python_variable_methods.cpp

  • 您可以在以下位置查找命名函数的文档:
    https://pytorch.org/docs/stable/torch.html?#torch.<FUNCTION-NAME>

    关于PyTorch 将运算符映射到函数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53370003/

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