tf.raw_ops.CTCBeamSearchDecoder
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.3/api_docs/python/tf/raw_ops/CTCBeamSearchDecoder