QuantWrapper

class torch.ao.quantization.QuantWrapper(module)

一个包装类，用于包装输入模块，添加 QuantStub 和 DeQuantStub，并在调用模块时调用 quant 和 dequant 模块。

量化工具函数使用它来添加 quant 和 dequant 模块。在 convert 函数之前，QuantStub 只会充当观察者，用于观察输入张量；在 convert 之后，QuantStub 将被替换为执行实际量化的 nnq.Quantize。对于 DeQuantStub 也是类似的。