Stay organized with collections Save and categorize content based on your preferences.

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.