decoder_beam_search_output {tfaddons}

Beam Search Decoder Output

Description

Beam Search Decoder Output

Usage

decoder_beam_search_output(scores, predicted_ids, parent_ids)

Arguments

scores

calculate the scores for each beam

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.

parent_ids

The parent ids of shape '[max_time, batch_size, beam_width]'.

Value

None