pytorch / 2 / generated / torch.ao.quantization.fake_quantize.fakequantizebase.html

FakeQuantizeBase

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

Base fake quantize module Any fake quantize implementation should derive from this class.

Concrete fake quantize module should follow the same API. In forward, they will update the statistics of the observed Tensor and fake quantize the input. They should also provide a calculate_qparams function that computes the quantization parameters given the collected statistics.

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.ao.quantization.fake_quantize.FakeQuantizeBase.html