QuantWrapper

类torch.ao.quantization.QuantWrapper(module)[源代码]

一个包装类，用于封装输入模块，并在其周围添加QuantStub和DeQuantStub。在调用该模块时，还会调用量化和去量化的模块。

这通过量化实用函数在convert函数之前添加量化和去量化的模块。在convert函数调用前，QuantStub仅作为观察器记录输入张量；而在convert之后，QuantStub将被替换为执行实际量化的nnq.Quantize。对于DeQuantStub也是如此。