Shortcuts

QuantWrapper

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

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

这是由量化实用函数用来添加量化和反量化模块的,在转换函数之前,量化存根只是一个观察者,它观察输入张量,在转换之后,量化存根将被替换为nnq.量化,它执行实际的量化。同样适用于反量化存根

优云智算