浮点功能

class torch.ao.nn.quantized.FloatFunctional

用于浮点操作的状态收集器类。

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

注意

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

示例：

>>> f_add = FloatFunctional()
>>> a = torch.tensor(3.0)
>>> b = torch.tensor(4.0)
>>> f_add.add(a, b)  # 等同于 ``torch.add(a, b)``

Valid operation names:

• 添加

• 猫

• 乘

• 添加_relu

• 添加标量

• mul_scalar