评价此页

FakeQuantizeBase#

class torch.ao.quantization.fake_quantize.FakeQuantizeBase[source]#

基础的伪量化模块。

所有伪量化实现都应继承自此类。

具体的伪量化模块应遵循相同的 API。在 `forward` 中,它们将更新观察到的 Tensor 的统计信息并对输入进行伪量化。它们还应提供一个 `calculate_qparams` 函数,该函数根据收集到的统计信息计算量化参数。