tf.compat.v1.nn.ctc_beam_search_decoder
Выполняет декодирование с помощью поиска по лучшим путям (beam search) на основе логарифмических вероятностей (logits), предоставленных на входе.
tf.compat.v1.nn.ctc_beam_search_decoder(
inputs, sequence_length, beam_width=100, top_paths=1, merge_repeated=True
)
Примечание:ctc_greedy_decoderявляется частным случаемctc_beam_search_decoderсtop_paths=1иbeam_width=1(но этот декодер быстрее для этого частного случая).
Если merge_repeated равно True, объединить повторяющиеся классы в выходных путях. Это означает, что если последовательные элементы в пути совпадают, то выводится только первый из них. То есть, когда последовательность A B B * B * B (где '*' - метка пробела), значение возврата будет:
-
A Bеслиmerge_repeated = True. -
A B B Bеслиmerge_repeated = False.
| Аргументы | |
|---|---|
inputs | 3-мерный float Tensor, размер [max_time x batch_size x num_classes]. Логарифмические вероятности (logits). |
sequence_length | 1-мерный int32 вектор, содержащий длины последовательностей, размер [batch_size]. |
beam_width | Целое число, скаляр >= 0 (ширина пучка поиска по лучшим путям). |
top_paths | Целое число, скаляр >= 0, <= beam_width (определяет размер выходных данных). |
merge_repeated | Булево. По умолчанию: True. |
| Возвращаемые значения | |
|---|---|
Кортеж (decoded, log_probabilities) , где | |
decoded | Список длиной top_paths, где decoded[j] является SparseTensor, содержащим декодированные выходные данные:
|
log_probability | Матрица float (batch_size x top_paths) содержащая логарифмы вероятностей последовательностей. |
© 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.3/api_docs/python/tf/compat/v1/nn/ctc_beam_search_decoder