tf.compat.v1.nn.quantized_conv2d
Вычисляет 2D свертку для квантованных тензоров входных и фильтрующих данных размерности 4.
tf.compat.v1.nn.quantized_conv2d(
input,
filter,
min_input,
max_input,
min_filter,
max_filter,
strides,
padding,
out_type=tf.dtypes.qint32,
dilations=[1, 1, 1, 1],
name=None
)
Входы представляют собой квантованные тензоры, где наименьшее значение соответствует реальному числу минимального значения, а наибольшее — максимального. Это означает, что вы можете интерпретировать квантованный вывод только таким же образом, учитывая возвращаемые минимальное и максимальное значения.
| Аргументы | |
|---|---|
input | A Tensor. Должен быть одним из следующих типов: qint8, quint8, qint32, qint16, quint16. |
filter | A Tensor. Должен быть одним из следующих типов: qint8, quint8, qint32, qint16, quint16. Размерность `input_depth` фильтра должна совпадать с размерностью `depth` входных данных. |
min_input | A Tensor типа float32. Вещественное значение, которое представляет наименьшее квантованное значение входных данных. |
max_input | A Tensor типа float32. Вещественное значение, которое представляет наибольшее квантованное значение входных данных. |
min_filter | A Tensor типа float32. Вещественное значение, которое представляет наименьшее квантованное значение фильтра. |
max_filter | A Tensor типа float32. Вещественное значение, которое представляет наибольшее квантованное значение фильтра. |
strides | Список ints. Шаг скользящего окна для каждой размерности тензора входных данных. |
padding | A string из: "SAME", "VALID". Тип алгоритма заполнения. |
out_type | Необязательный tf.DType из: tf.qint8, tf.quint8, tf.qint32, tf.qint16, tf.quint16. По умолчанию tf.qint32. |
dilations | Необязательный список ints. По умолчанию [1, 1, 1, 1]. 1-мерный тензор длиной 4. Коэффициент расширения для каждой размерности тензора input. Если установлено значение k > 1, между каждым элементом фильтра по этой размерности будет k-1 пропущенных ячеек. Порядок размерностей определяется значением data_format, подробности см. выше. Расширения по размерностям пакетных и глубины должны быть 1. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Кортеж объектов Tensor (выход, min_выход, max_выход). | |
output | A Tensor типа out_type. |
min_output | A Tensor типа float32. |
max_output | A Tensor типа 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/versions/r2.9/api_docs/python/tf/compat/v1/nn/quantized_conv2d