tfa.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