FakeQuantizeBase

类 torch.ao.quantization.fake_quantize.FakeQuantizeBase

基础伪量化模块。

基础伪量化模块。任何伪量化实现都应从此类派生。

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