QuantWrapper

class torch.ao.quantization.QuantWrapper(module)

一個包裝類別，用於包裝輸入模組，並新增 QuantStub 和 DeQuantStub，且用呼叫 quant 和 dequant 模組的方式來包圍對模組的呼叫。

此類別由 quantization 工具函數使用，以在 convert 函數之前新增 quant 和 dequant 模組。QuantStub 僅作為觀察者，觀察輸入張量。在 convert 之後，QuantStub 將被替換為 nnq.Quantize，執行實際的量化。 DeQuantStub 也是類似的情況。