FakeQuantizeBase

类torch.ao.quantization.fake_quantize.FakeQuantizeBase[源代码]

基本的假量化模块。

基础假量化模块：任何假量化实现都应继承此类。

具体的伪量化模块应该遵循相同的API。在前向传播过程中，它们会更新观察到的张量的统计信息，并对输入进行伪量化。还应提供一个calculate_qparams函数来根据收集到的统计信息计算量化参数。