tf.nn.ctc_greedy_decoder
| View source on GitHub |
Performs greedy decoding on the logits given in input (best path).
tf.nn.ctc_greedy_decoder(
inputs, sequence_length, merge_repeated=True
)
Note: Regardless of the value of merge_repeated, if the maximum index of a given time and batch corresponds to the blank index (num_classes - 1), no new element is emitted.
If merge_repeated is True, merge repeated classes in output. This means that if consecutive logits' maximum indices are the same, only the first of these is emitted. The sequence A B B * B * B (where '*' is the blank label) becomes
-
A B B Bifmerge_repeated=True. -
A B B B Bifmerge_repeated=False.
| Args | |
|---|---|
inputs | 3-D float Tensor sized [max_time, batch_size, num_classes]. The logits. |
sequence_length | 1-D int32 vector containing sequence lengths, having size [batch_size]. |
merge_repeated | Boolean. Default: True. |
| Returns | |
|---|---|
A tuple (decoded, neg_sum_logits) where | |
decoded | A single-element list. decoded[0] is an SparseTensor containing the decoded outputs s.t.:
|
neg_sum_logits | A float matrix (batch_size x 1) containing, for the sequence found, the negative of the sum of the greatest logit at each timeframe. |
© 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/nn/ctc_greedy_decoder