Beam Search Decoder Output
decoder_beam_search_output(scores, predicted_ids, parent_ids)
calculate the scores for each beam
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.
The parent ids of shape `[max_time, batch_size, beam_width]`.
None