Warning: This project is deprecated. TensorFlow Addons has stopped development, The project will only be providing minimal maintenance releases until May 2024. See the full announcement here or on github.

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.