评价此页

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