FloatFunctional

类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)``

允许的操作名称：

• 添加

• 猫

• mul

• 添加ReLU

• 添加标量值

• 标量乘法