Performs beam search decoding on the logits given in input.
tf.raw_ops.CTCBeamSearchDecoder( inputs, sequence_length, beam_width, top_paths, merge_repeated=True, name=None )
A note about the attribute merge_repeated: For the beam search decoder, this means that if consecutive entries in a beam are the same, only the first of these is emitted. That is, when the top path is "A B B B B", "A B" is returned if merge_repeated = True but "A B B B B" is returned if merge_repeated = False.
Args | |
---|---|
inputs | A Tensor . Must be one of the following types: float32 , float64 . 3-D, shape: (max_time x batch_size x num_classes) , the logits. |
sequence_length | A Tensor of type int32 . A vector containing sequence lengths, size (batch) . |
beam_width | An int that is >= 1 . A scalar >= 0 (beam search beam width). |
top_paths | An int that is >= 1 . A scalar >= 0, <= beam_width (controls output size). |
merge_repeated | An optional bool . Defaults to True . If true, merge repeated classes in output. |
name | A name for the operation (optional). |
Returns | |
---|---|
A tuple of Tensor objects (decoded_indices, decoded_values, decoded_shape, log_probability). | |
decoded_indices | A list of top_paths Tensor objects with type int64 . |
decoded_values | A list of top_paths Tensor objects with type int64 . |
decoded_shape | A list of top_paths Tensor objects with type int64 . |
log_probability | A Tensor . Has the same type as inputs . |
© 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/raw_ops/CTCBeamSearchDecoder