tf.raw_ops.CTCBeamSearchDecoder
Выполняет декодирование с поиском по лучу для переданных на вход логитов.
tf.raw_ops.CTCBeamSearchDecoder(
inputs,
sequence_length,
beam_width,
top_paths,
merge_repeated=True,
name=None
)
Примечание об атрибуте merge_repeated: для декодера с поиском по лучу это означает, что если соседние элементы в луче совпадают, выводится только первый из них. То есть, если лучший путь — "A B B B B", при merge_repeated = True возвращается "A B", а при merge_repeated = False — "A B B B B".
| Аргументы | |
|---|---|
inputs | A Tensor. Должен иметь один из следующих типов: float32, float64. 3-D, форма: (max_time x batch_size x num_classes), логиты. |
sequence_length | A Tensor типа int32. Вектор, содержащий длины последовательностей, размер (batch). |
beam_width | An int типа >= 1. Скаляр >= 0 (ширина луча при поиске по лучу). |
top_paths | An int типа >= 1. Скаляр >= 0, <= beam_width (задает размер выходных данных). |
merge_repeated | Необязательный bool. По умолчанию — True. Если значение истинно, объединяет повторяющиеся классы в выходных данных. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Кортеж объектов Tensor (decoded_indices, decoded_values, decoded_shape, log_probability). | |
decoded_indices | Список объектов top_paths Tensor типа int64. |
decoded_values | Список объектов top_paths Tensor типа int64. |
decoded_shape | Список объектов top_paths Tensor типа int64. |
log_probability | Значение Tensor. Имеет тот же тип, что и inputs. |
© 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/raw_ops/CTCBeamSearchDecoder