tfa.seq2seq.FinalBeamSearchDecoderOutput

View source on GitHub

Class FinalBeamSearchDecoderOutput

Final outputs returned by the beam search after all decoding is

finished.

Args:

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

__new__

@staticmethod
__new__(
    _cls,
    predicted_ids,
    beam_search_decoder_output
)

Create new instance of FinalBeamDecoderOutput(predicted_ids, beam_search_decoder_output)

Properties

predicted_ids

beam_search_decoder_output