FloatFunctional

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)  # Equivalent to ``torch.add(a, b)``

有效操作名称

• add

• cat

• mul

• add_relu

• add_scalar

• mul_scalar