QuantWrapper ¬

类 torch.ao.quantization.QuantWrapper(module)[source][source] ¬

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

这是由量化实用函数使用的，用于添加量化和去量化模块，在转换函数 QuantStub 中，它只是观察输入张量，转换后，QuantStub 将替换为 nnq.Quantize 以执行实际的量化。对于 DeQuantStub 也是类似。