QFunctional

class torch.ao.nn.quantized.QFunctional

量化操作的包裝類別。

此類別的實例可以用來代替 torch.ops.quantized 前綴。 請參閱下面的範例用法。

注意

此類別不提供 forward hook。您必須改用底層函式之一 (例如 add)。

範例

>>> q_add = QFunctional()
>>> a = torch.quantize_per_tensor(torch.tensor(3.0), 1.0, 0, torch.qint32)
>>> b = torch.quantize_per_tensor(torch.tensor(4.0), 1.0, 0, torch.qint32)
>>> q_add.add(a, b)  # Equivalent to ``torch.ops.quantized.add(a, b, 1.0, 0)``

有效的操作名稱

• add

• cat

• mul

• add_relu

• add_scalar

• mul_scalar