QuantWrapper

class torch.ao.quantization.QuantWrapper(module) [источник]

Класс-обёртка, оборачивающий входной модуль, добавляющий QuantStub и DeQuantStub и обрамляющий вызов модуля вызовами модулей квантования и деквантования.

Этот класс используется вспомогательными функциями quantization для добавления модулей квантования и деквантования. До вызова функции convert QuantStub будет только наблюдателем: он наблюдает за входным тензором. После вызова функции convert QuantStub будет заменён на nnq.Quantize, который выполняет фактическое квантование. Аналогично и для DeQuantStub.