TensorFlow 2.0 Beta is available Learn more

tf.contrib.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.

Properties

predicted_ids

beam_search_decoder_output