QFunctional#
- class torch.ao.nn.quantized.QFunctional[source]#
用于量化操作的包装类。
此类的实例可以替代
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