tf.nn.ctc_beam_search_decoder
| Просмотреть исходный код на GitHub |
Выполняет декодирование с использованием лучевого поиска для логарифмических вероятностей, заданных на входе.
tf.nn.ctc_beam_search_decoder(
inputs, sequence_length, beam_width=100, top_paths=1
)
Примечание: Хотя в общем случае жадный поиск является частным случаем лучевого поиска сtop_paths=1иbeam_width=1,ctc_beam_search_decoderотличается отctc_greedy_decoderв обработке пробелов при вычислении вероятности последовательности:
-
ctc_beam_search_decoderрассматривает пробелы как завершение последовательности -
ctc_greedy_decoderрассматривает пробелы как обычные элементы
| Аргументы | |
|---|---|
inputs | 3-мерный float Tensor, размер [max_time, batch_size, num_classes]. Логарифмические вероятности. |
sequence_length | 1-мерный int32 вектор, содержащий длины последовательностей, размер [batch_size]. |
beam_width | Целое скалярное значение ≥ 0 (ширина луча для лучевого поиска). |
top_paths | Целое скалярное значение ≥ 0, ≤ beam_width (управляет размером вывода). |
| Возвращаемые значения | |
|---|---|
Кортеж (decoded, log_probabilities) , где | |
decoded | Список длиной top_paths, где decoded[j] является SparseTensor, содержащим декодированные выходные данные:
|
log_probability | Матрица float [batch_size, top_paths] , содержащая логарифмические вероятности последовательностей. |
© 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/nn/ctc_beam_search_decoder