MovingAveragePerChannelMinMaxObserver

class torch.ao.quantization.observer.MovingAveragePerChannelMinMaxObserver(averaging_constant=0.01, ch_axis=0, dtype=torch.quint8, qscheme=torch.per_channel_affine, reduce_range=False, quant_min=None, quant_max=None, eps=1.1920928955078125e-07, is_dynamic=False, **kwargs) [исходный код]

Модуль-наблюдатель для вычисления параметров квантования на основе скользящих минимальных и максимальных значений для каждого канала.

Этот наблюдатель использует статистику минимальных и максимальных значений тензора для вычисления параметров квантования для каждого канала. Модуль сохраняет скользящие минимум и максимум входящих тензоров и использует эту статистику для вычисления параметров квантования.

Параметры:

• averaging_constant – Константа усреднения для минимальных и максимальных значений.
• ch_axis – Ось каналов
• dtype – Тип квантованных данных
• qscheme – Используемая схема квантования
• reduce_range – Уменьшает диапазон типа квантованных данных на 1 бит
• quant_min – Минимальное значение квантования. Если не задано, используется конфигурация для 8-битного представления.
• quant_max – Максимальное значение квантования. Если не задано, используется конфигурация для 8-битного представления.
• eps (Tensor) – Значение эпсилон для float32. По умолчанию равно torch.finfo(torch.float32).eps.

Параметры квантования вычисляются так же, как в MovingAverageMinMaxObserver, за исключением того, что скользящие минимальные и максимальные значения хранятся отдельно для каждого канала. Поэтому масштабы и нулевые точки также вычисляются отдельно для каждого канала.

Примечание

Если скользящий минимум равен скользящему максимуму, масштабы и нулевые точки устанавливаются в 1.0 и 0.