FloatFunctional

class torch.ao.nn.quantized.FloatFunctional

用於浮點運算的狀態收集器類別。

此類別的實例可以取代 torch. 前綴來進行某些運算。 請參閱下面的範例用法。

注意

此類別不提供 forward hook。 相反，您必須使用其中一個底層函數 (例如 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