tf.quantization.quantized_concat
Выполняет конкатенацию квантованных тензоров по одному измерению.
tf.quantization.quantized_concat(
concat_dim: Annotated[Any, _atypes.Int32],
values: Annotated[List[Any], TV_QuantizedConcat_T],
input_mins: Annotated[List[Any], _atypes.Float32],
input_maxes: Annotated[List[Any], _atypes.Float32],
name=None
)
| Аргументы | |
|---|---|
concat_dim | Численный тип Tensor. 0-D. Измерение для конкатенации. Должно быть в диапазоне [0, ранг(значения)). |
values | Список из как минимум 2 объектов Tensor с одинаковым типом. Квантованные тензоры для конкатенации. Их ранг и типы должны совпадать, а размеры во всех измерениях, кроме concat_dim, также должны совпадать. |
input_mins | Список с длиной, равной длине values, объектов Tensor типа float32. Минимальные скалярные значения для каждого из входных тензоров. |
input_maxes | Список с длиной, равной длине values, объектов Tensor типа float32. Максимальные скалярные значения для каждого из входных тензоров. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Кортеж из объектов Tensor (выход, выходной_мин, выходной_макс). | |
output | Квантованный тензор Tensor. Имеет тот же тип, что и values. |
output_min | Список Tensor типа float32. |
output_max | Список 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/api_docs/python/tf/quantization/quantized_concat