ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.contrib.seq2seq.FinalBeamSearchDecoderOutput

View source on GitHub

Final outputs returned by the beam search after all decoding is finished.

predicted_ids The final prediction. A tensor of shape [batch_size, T, beam_width] (or [T, batch_size, beam_width] if output_time_major is True). Beams are ordered from best to worst.
beam_search_decoder_output An instance of BeamSearchDecoderOutput that describes the state of the beam search.

predicted_ids

beam_search_decoder_output