gather_tree

step_ids

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

parent_ids

get max_sequence_length across all beams for each batch.

max_sequence_lengths

`int32` scalar, the token that marks end of decoding.

end_token

'TensorFlow SIG Addons' <https://www.tensorflow.org/addons> is a repository
of community contributions that conform to well-established API patterns,
but implement new functionality not available in core 'TensorFlow'.
'TensorFlow' natively supports a large number of operators, layers, metrics,
losses, optimizers, and more. However, in a fast moving field like Machine Learning,
there are many interesting new developments that cannot be integrated into
core 'TensorFlow' (because their broad applicability is not yet clear, or
it is mostly used by a smaller subset of the community).

gather_tree: Gather tree

Description

Usage

Arguments

Value