QuantWrapper

class torch.ao.quantization.QuantWrapper(module)

一个包装类，它包装输入模块，添加 QuantStub 和 DeQuantStub，并将对模块的调用与 Quant 和 Dequant 模块的调用包围起来。

这由 quantization 实用函数用于添加 Quant 和 Dequant 模块，在 convert 函数之前，QuantStub 只是一个观察者，它观察输入张量，在 convert 之后，QuantStub 将被交换为 nnq.Quantize，执行实际的量化。DeQuantStub 同理。