Save the date! Google I/O returns May 18-20 Register now


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.