QFunctional

torch.ao.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)``
允许的操作名称:
  • 添加

  • mul

  • 添加ReLU

  • 添加标量值

  • 标量乘法

本页目录