tensorflow::ops::QuantizeV2
#include <array_ops.h>Преобразуйте тензор «входа» типа float в тензор «выхода» типа T.
Краткое содержание
[min_range, max_range] — это скалярные числа с плавающей запятой, определяющие диапазон входных данных. Атрибут 'mode' точно определяет, какие вычисления используются для преобразования значений с плавающей запятой в их квантованные эквиваленты. Атрибут 'round_mode' определяет, какой алгоритм округления используется при округлении значений с плавающей запятой до их квантованных эквивалентов.
В режиме 'MIN_COMBINED' каждое значение тензора будет подвергаться следующим изменениям:
out[i] = (in[i] - min_range) * range(T) / (max_range - min_range) if T == qint8: out[i] -= (range(T) + 1) / 2.0
здесь range(T) = numeric_limits
Пример режима MIN_COMBINED
Предположим, входные данные имеют тип float и возможный диапазон значений [0.0, 6.0], а выходные данные — тип quint8 ([0, 255]). Значения min_range и max_range должны быть указаны как 0.0 и 6.0. Квантование из типа float в quint8 приведет к умножению каждого входного значения на 255/6 и преобразованию в тип quint8.
Если тип выходных данных — qint8 ([-128, 127]), то перед преобразованием типа из каждого значения будет дополнительно вычтено 128, чтобы диапазон значений совпадал с диапазоном qint8.
Если установлен режим '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 = num_discrete_values / range quantized = round(input * range_scale) - round(range_min * range_scale) + numeric_limits::min() quantized = max(quantized, numeric_limits ::min()) quantized = min(quantized, numeric_limits ::max())
Главное отличие от MIN_COMBINED заключается в том, что здесь сначала округляется минимальный диапазон, а затем он вычитается из округленного значения. В случае с MIN_COMBINED вводится небольшая погрешность, поскольку повторные итерации квантования и деквантования приводят к всё большей и большей ошибке.
Пример масштабируемого режима
Режим SCALED соответствует подходу квантования, используемому в QuantizeAndDequantize{V2|V3} .
Если установлен режим SCALED , квантование выполняется путем умножения каждого входного значения на коэффициент масштабирования. Коэффициент масштабирования определяется на основе значений min_range и max_range и устанавливается таким образом, чтобы диапазон от min_range до max_range был представим в пределах значений типа T.
const int min_T = std::numeric_limits::min(); const int max_T = std::numeric_limits ::max(); const float max_float = std::numeric_limits ::max();
const float scale_factor_from_min_side = (min_T * min_range > 0) ? min_T / min_range : max_float; const float scale_factor_from_max_side = (max_T * max_range > 0) ? max_T / max_range : max_float;
const float scale_factor = std::min(scale_factor_from_min_side, scale_factor_from_max_side);
Далее мы используем параметр scale_factor для корректировки параметров min_range и max_range следующим образом:
min_range = min_T / scale_factor;
max_range = max_T / scale_factor;Например, если T = qint8, и изначально min_range = -10, а max_range = 9, мы сравним -128/-10.0 = 12.8 с 127/9.0 = 14.11 и установим scaling_factor = 12.8. В этом случае min_range останется -10, но max_range будет скорректирован до 127 / 12.8 = 9.921875.
Таким образом, мы будем квантовать входные значения в диапазоне от (-10, 9,921875) до (-128, 127).
Теперь входной тензор можно квантовать, ограничив значения диапазоном от min_range до max_range , а затем умножив на scale_factor следующим образом:
result = round(min(max_range, max(min_range, input)) * scale_factor)
Скорректированные min_range и max_range возвращаются в качестве выходных данных 2 и 3 этой операции. Эти выходные данные следует использовать в качестве диапазона для любых дальнейших вычислений.
атрибут narrow_range (логическое значение)
Если это так, мы не используем минимальное квантованное значение. То есть для int8 квантованный выход будет ограничен диапазоном от -127 до 127 вместо полного диапазона от -128 до 127. Это предусмотрено для совместимости с некоторыми бэкэндами вывода. (Применяется только к режиму SCALED)
атрибут оси (целое число)
Необязательный атрибут axis может указывать индекс размерности входного тензора, так что диапазоны квантования будут вычисляться и применяться отдельно для каждого среза тензора вдоль этой размерности. Это полезно для квантования по каждому каналу.
Если указана ось, то min_range и max_range.
Если axis = None, квантование каждого тензора выполняется как обычно.
атрибут ensure_minimum_range (float)
Гарантирует, что минимальный диапазон квантования будет не меньше этого значения. Устаревшее значение по умолчанию для этого параметра равно 0,01, но настоятельно рекомендуется установить его равным 0 для новых применений.
Аргументы:
- scope: Объект Scope
- min_range: Минимальное значение диапазона квантования. Это значение может быть скорректировано операцией в зависимости от других параметров. Скорректированное значение записывается в
output_min. Если указан атрибутaxis, это должен быть одномерный тензор, размер которого соответствует размерностиaxisвходного и выходного тензоров. - max_range: Максимальное значение диапазона квантования. Это значение может быть скорректировано операцией в зависимости от других параметров. Скорректированное значение записывается в
output_max. Если указан атрибутaxis, это должен быть одномерный тензор, размер которого соответствует размерностиaxisвходного и выходного тензоров.
Возвращает:
-
Outputданные: квантованные данные, полученные из входных данных в виде чисел с плавающей запятой. -
Outputoutput_min: Минимальный диапазон квантования, используемый для обрезки входных значений перед масштабированием и округлением до квантованных значений. Если указан атрибутaxis, это будет одномерный тензор, размер которого соответствует размерностиaxisвходного и выходного тензоров. -
Outputпараметр output_max: Максимальный диапазон квантования, используемый для обрезки входных значений перед масштабированием и округлением до квантованных значений. Если указан атрибутaxis, это будет одномерный тензор, размер которого соответствует размерностиaxisвходного и выходного тензоров.
Конструкторы и Деструкторы | |
|---|---|
QuantizeV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input min_range, :: tensorflow::Input max_range, DataType T) | |
QuantizeV2 (const :: tensorflow::Scope & scope, :: tensorflow::Input input, :: tensorflow::Input min_range, :: tensorflow::Input max_range, DataType T, const QuantizeV2::Attrs & attrs) |
Общедоступные атрибуты | |
|---|---|
operation | |
output | |
output_max | |
output_min | |
Открытые статические функции | |
|---|---|
Axis (int64 x) | |
EnsureMinimumRange (float x) | |
Mode (StringPiece x) | |
NarrowRange (bool x) | |
RoundMode (StringPiece x) | |
Структуры | |
|---|---|
| tensorflow::ops::QuantizeV2::Attrs | Дополнительные параметры установки атрибутов для QuantizeV2 . |
Общедоступные атрибуты
операция
Operation operation
выход
::tensorflow::Output output
output_max
::tensorflow::Output output_max
output_min
::tensorflow::Output output_min
Общественные функции
QuantizeV2
QuantizeV2( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input min_range, ::tensorflow::Input max_range, DataType T )
QuantizeV2
QuantizeV2( const ::tensorflow::Scope & scope, ::tensorflow::Input input, ::tensorflow::Input min_range, ::tensorflow::Input max_range, DataType T, const QuantizeV2::Attrs & attrs )
Открытые статические функции
Ось
Attrs Axis( int64 x )
EnsureMinimumRange
Attrs EnsureMinimumRange( float x )
Режим
Attrs Mode( StringPiece x )
Узкий диапазон
Attrs NarrowRange( bool x )
Круглый режим
Attrs RoundMode( StringPiece x )