快捷方式

FloatFunctional

class torch.ao.nn.quantized.FloatFunctional[source]

浮点运算的状态收集器类。

此类的实例可以代替 torch. 前缀用于某些操作。请参见下面的示例用法。

注意

此类不提供 forward 钩子。相反,您必须使用其中一个底层函数(例如 add)。

示例

>>> f_add = FloatFunctional()
>>> a = torch.tensor(3.0)
>>> b = torch.tensor(4.0)
>>> f_add.add(a, b)  # Equivalent to ``torch.add(a, b)``
有效的操作名称
  • add

  • cat

  • mul

  • add_relu

  • add_scalar

  • mul_scalar

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得解答

查看资源