添加量化与去量化的操作

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

如果叶级子模块具有有效的qconfig,则使用QuantWrapper将其包裹起来。需要注意的是,此函数会直接修改模块的子模块,并可能返回一个包装了输入模块的新模块。

参数
  • module - 输入模块,包含所有叶子模块的 qconfig 属性

  • 量化(我们希望实现的)–

返回值

要么是对现有模块进行就地修改,并使用基于 qconfigQuantWrapper 包装其子模块,要么是创建一个新的 QuantWrapper 模块来封装输入模块。后一种情况仅在输入模块为叶模块且需要对其进行量化时发生。

本页目录