Python PyTorch FloatFunctional用法及代碼示例

本文簡要介紹python語言中 torch.nn.quantized.FloatFunctional 的用法。

用法:

class torch.nn.quantized.FloatFunctional

用於浮點操作的狀態Collector類。

對於某些操作，可以使用此類的實例代替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