|View source on GitHub|
Calculates the full beams for
tfa.seq2seq.gather_tree_from_array( t, parent_ids, sequence_length )
t: A stacked
Tensors of shape
[batch_size, beam_width, s]or
[batch_size * beam_width, s]where
sis the depth shape.
parent_ids: The parent ids of shape
[max_time, batch_size, beam_width].
sequence_length: The sequence length of shape
Tensor which is a stacked
TensorArray of the same size and type as
t and where beams are sorted in each
Tensor according to