tf.quantization.fake_quant_with_min_max_vars_per_channel

Искусственно квантует тензор 'inputs' типа float по каналам.

tf.quantization.fake_quant_with_min_max_vars_per_channel(
    inputs: Annotated[Any, _atypes.Float32],
    min: Annotated[Any, _atypes.Float32],
    max: Annotated[Any, _atypes.Float32],
    num_bits: int = 8,
    narrow_range: bool = False,
    name=None
) -> Annotated[Any, _atypes.Float32]

Искусственно квантует тензор типа float по каналам, имеющий одну из следующих форм: [d], [b, d] [b, h, w, d], используя по-канальные float значения min и max формы [d], в тензор с той же формой, что и inputs.

Атрибуты

• [min; max] определяют диапазон ограничения для данных inputs.
• inputs значения квантуются в диапазон квантования ([0; 2^num_bits - 1], если narrow_range равно false и [1; 2^num_bits - 1], если оно равно true), затем деквантуются и выводятся как float в интервале [min; max].
• num_bits — разрядность квантования; от 2 до 16 включительно.

Перед квантованием значения min и max корректируются с использованием следующего логики. Рекомендуется иметь min <= 0 <= max. Если 0 не входит в диапазон значений, поведение может быть непредсказуемым:

• Если 0 < min < max: min_adj = 0 и max_adj = max - min.
• Если min < max < 0: min_adj = min - max и max_adj = 0.
• Если min <= 0 <= max: scale = (max - min) / (2^num_bits - 1), min_adj = scale * round(min / scale) и max_adj = max + min_adj - min.

Эта операция имеет градиент и, следовательно, позволяет обучать значения min и max.