View source on GitHub |
Performs beam search decoding on the logits given in input.
tf.nn.ctc_beam_search_decoder_v2(
inputs, sequence_length, beam_width=100, top_paths=1
)
Args | |
---|---|
inputs
|
3-D float Tensor , size [max_time, batch_size, num_classes] .
The logits.
|
sequence_length
|
1-D int32 vector containing sequence lengths, having size
[batch_size] .
|
beam_width
|
An int scalar >= 0 (beam search beam width). |
top_paths
|
An int scalar >= 0, <= beam_width (controls output size). |
Returns | |
---|---|
A tuple (decoded, log_probabilities) where
|
|
decoded
|
A list of length top_paths, where decoded[j]
is a SparseTensor containing the decoded outputs:
|
log_probability
|
A float matrix [batch_size, top_paths] containing
sequence log-probabilities.
|