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