Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

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