add_quant_dequant

class torch.ao.quantization.add_quant_dequant(module)

如果葉子子模組具有有效的 qconfig，則將其包裝在 QuantWrapper 中。請注意，此函式將原地修改模組的子模組，並且還可以返回一個新模組來包裝輸入模組。

引數

• module – 帶有所有葉子模組 qconfig 屬性的輸入模組

• quantize (我們要量化的) –

返回

根據 qconfig 返回原地修改的模組（其子模組已包裝在 QuantWrapper 中），或返回一個包裝輸入模組的新 QuantWrapper 模組；後一種情況僅發生在輸入模組是葉子模組且我們想要對其進行量化時。