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

tfa.seq2seq.FinalBeamSearchDecoderOutput

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 tfa.seq2seq.BeamSearchDecoderOutput that describes the state of the beam search.