评价此页

QuantWrapper#

class torch.ao.quantization.QuantWrapper(module)[source]#

一个包装输入模块的包装器类,添加 QuantStub 和 DeQuantStub,并将对模块的调用用对量化和反量化模块的调用包围起来。

这由 `quantization` 实用函数使用,用于在 `convert` 函数之前添加量化和反量化模块。`QuantStub` 最初是一个观察器,它观察输入张量;在 `convert` 之后,`QuantStub` 将被替换为 `nnq.Quantize`,后者执行实际的量化。DeQuantStub 也是类似的。