View source on GitHub
|
Final outputs returned by the beam search after all decoding is finished.
tfa.seq2seq.FinalBeamSearchDecoderOutput(
predicted_ids, beam_search_decoder_output
)
Attributes | |
|---|---|
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.
|