Beam Search Decoder Output
decoder_beam_search_output(scores, predicted_ids, parent_ids)
| 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]`. |
None