评价此页

add_quant_dequant#

class torch.ao.quantization.add_quant_dequant(module)[源代码]#

将具有有效 qconfig 的叶子子模块包装在 QuantWrapper 中。注意此函数将就地修改模块的子模块,并且还可以返回一个包装输入模块的新模块。

参数:
  • module – 输入模块,所有叶子模块都具有 qconfig 属性

  • quantize (我们想) –

返回:

根据 qconfig 将子模块包装在 QuantWrapper 中的就地修改后的模块,或者包装输入模块的新 QuantWrapper 模块。后一种情况仅发生在输入模块是叶子模块且我们想对其进行量化时。