浮点功能 ¶

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

有效的操作名称：

• 添加

• 猫

• 多

• 添加 ReLU

• 添加标量

• 多标量