tf.quantization.fake_quant_with_min_max_vars
Имитирует квантование тензора 'inputs' типа float с помощью глобальных скаляров типа float
tf.quantization.fake_quant_with_min_max_vars(
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 с помощью глобальных скаляров float min и max в тензор той же формы, что и inputs.
Атрибуты
-
[min; max]определяют диапазон ограничения данныхinputs. -
Значения
inputsквантуются в диапазон квантования ([0; 2^num_bits - 1], когдаnarrow_rangeравно false и[1; 2^num_bits - 1], когда оно равно true), а затем деквантуются и выводятся в виде чисел с плавающей запятой в диапазоне[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.
| Аргументы | |
|---|---|
inputs | Тензор типа float32. |
min | Тензор типа float32. |
max | Тензор типа float32. |
num_bits | Необязательный параметр. По умолчанию 8. |
narrow_range | Необязательный параметр. По умолчанию False. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Тензор типа float32. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/quantization/fake_quant_with_min_max_vars