Python PyTorch QFunctional用法及代码示例

本文简要介绍python语言中 torch.nn.quantized.QFunctional 的用法。

用法:

class torch.nn.quantized.QFunctional

量化操作的包装类。

可以使用此类的实例来代替torch.ops.quantized 前缀。请参阅下面的示例用法。

注意

此类不提供forward 挂钩。相反，您必须使用基础函数之一(例如 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