tf.quantize_and_dequantize_v4
Возвращает градиент QuantizeAndDequantizeV4.
tf.quantize_and_dequantize_v4(
input, input_min, input_max, signed_input=True, num_bits=8, range_given=False,
round_mode='HALF_TO_EVEN', narrow_range=False, axis=-1, name=None
)
Это почти идентично QuantizeAndDequantizeV2, за исключением того, что оно возвращает градиент 1 для входных данных, которые находятся в диапазоне квантования, или 0 в противном случае.
| Аргументы | |
|---|---|
input | A Tensor. Должен быть одним из следующих типов: bfloat16, half, float32, float64. |
input_min | A Tensor. Должен иметь тот же тип, что и input. |
input_max | A Tensor. Должен иметь тот же тип, что и input. |
signed_input | Необязательный bool. По умолчанию True. |
num_bits | Необязательный int. По умолчанию 8. |
range_given | Необязательный bool. По умолчанию False. |
round_mode | Необязательный string из: "HALF_TO_EVEN", "HALF_UP". По умолчанию "HALF_TO_EVEN". |
narrow_range | Необязательный bool. По умолчанию False. |
axis | Необязательный int. По умолчанию -1. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
A Tensor. Имеет тот же тип, что и input. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/quantize_and_dequantize_v4