tf.quantization.dequantize
Деквантовать тензор 'input' в тензор типа float или bfloat16.
tf.quantization.dequantize(
input,
min_range,
max_range,
mode='MIN_COMBINED',
name=None,
axis=None,
narrow_range=False,
dtype=tf.dtypes.float32
)
[min_range, max_range] — скалярные значения типа float, задающие диапазон выходных данных. Атрибут 'mode' определяет, какие именно вычисления используются для преобразования значений float в соответствующие квантованные значения.
В режиме 'MIN_COMBINED' каждое значение тензора обрабатывается следующим образом:
if T == qint8: in[i] += (range(T) + 1)/ 2.0 out[i] = min_range + (in[i]* (max_range - min_range) / range(T))
здесь range(T) = numeric_limits<T>::max() - numeric_limits<T>::min()
Пример режима MIN_COMBINED
Если входные данные получены из QuantizedRelu6, выходной тип — quint8 (диапазон 0–255), но возможный диапазон QuantizedRelu6 составляет 0–6. Поэтому значения min_range и max_range равны 0.0 и 6.0. Операция Dequantize для quint8 преобразует каждое значение в float и умножает его на 6 / 255. Обратите внимание: если quantizedtype имеет тип qint8, перед преобразованием операция дополнительно прибавляет 128 к каждому значению.
Если выбран режим 'MIN_FIRST', используется следующий подход:
num_discrete_values = 1 << (# of bits in T) range_adjust = num_discrete_values / (num_discrete_values - 1) range = (range_max - range_min) * range_adjust range_scale = range / num_discrete_values const double offset_input = static_cast<double>(input) - lowest_quantized; result = range_min + ((input - numeric_limits<T>::min()) * range_scale)
Если выбран режим SCALED, деквантование выполняется умножением каждого входного значения на scaling_factor. (Таким образом, входное значение 0 всегда преобразуется в 0.0.)
Значение scaling_factor определяется на основе min_range, max_range и narrow_range таким образом, чтобы обеспечить совместимость с QuantizeAndDequantize{V2|V3} и QuantizeV2, с помощью следующего алгоритма:
const int min_expected_T = std::numeric_limits<T>::min() +
(narrow_range ? 1 : 0);
const int max_expected_T = std::numeric_limits<T>::max();
const float max_expected_T = std::numeric_limits<float>::max();
const float scale_factor =
(std::numeric_limits<T>::min() == 0) ? (max_range / max_expected_T)
: std::max(min_range / min_expected_T,
max_range / max_expected_T);
| Аргументы | |
|---|---|
input | Tensor. Должен иметь один из следующих типов: qint8, quint8, qint32, qint16, quint16. |
min_range | Tensor типа float32. Минимальное скалярное значение, которое может быть получено для входных данных. |
max_range | Tensor типа float32. Максимальное скалярное значение, которое может быть получено для входных данных. |
mode | Необязательный string из: "MIN_COMBINED", "MIN_FIRST", "SCALED". Значение по умолчанию — "MIN_COMBINED". |
narrow_range | Необязательный bool. Значение по умолчанию — False. |
axis | Необязательный int. Значение по умолчанию — -1. |
dtype | Необязательный tf.DType из: tf.bfloat16, tf.float32. Значение по умолчанию — tf.float32. Тип выходного тензора. В настоящее время Dequantize поддерживает типы float и bfloat16. Если для 'dtype' задано значение 'bfloat16', поддерживается только режим 'MIN_COMBINED'. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Tensor типа dtype. |
© 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/dequantize