QuantWrapper

class torch.ao.quantization.QuantWrapper(module)

一個包裝類，用於包裝輸入模組，新增 QuantStub 和 DeQuantStub，並在呼叫模組時呼叫 quant 和 dequant 模組。

量化工具函式使用它來新增 quant 和 dequant 模組。在 convert 函式之前，QuantStub 只會充當觀察者，用於觀察輸入張量；在 convert 之後，QuantStub 將被替換為執行實際量化的 nnq.Quantize。對於 DeQuantStub 也是類似的。