decoder_final_beam_search_output: Final Beam Search Decoder Output

In henry090/tfaddons: Interface to 'TensorFlow SIG Addons'

Final Beam Search Decoder Output

Description

Final outputs returned by the beam search after all decoding is finished.

Usage

decoder_final_beam_search_output(predicted_ids, beam_search_decoder_output)

Arguments

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.

Value

None

henry090/tfaddons documentation built on April 7, 2022, 11:27 p.m.