tf.contrib.seq2seq.gather_tree
Вычисляет полные лучи по идентификаторам на каждом шаге и идентификаторам родительских лучей.
tf.contrib.seq2seq.gather_tree(
step_ids, parent_ids, max_sequence_lengths, end_token, name=None
)
В CPU, если обнаружен идентификатор родителя за пределами границ, возвращается ошибка. В GPU, если обнаружен идентификатор родителя за пределами границ, соответствующее значение выходного параметра устанавливается в -1, и выполнение для этого луча прерывается.
Для данного луча, после шага во времени, содержащего первый декодированный end_token все значения заполняются end_token.
| Аргументы | |
|---|---|
step_ids | A Tensor. Должен быть одного из следующих типов: int32. [max_time, batch_size, beam_width]. |
parent_ids | A Tensor. Должен иметь тот же тип, что и step_ids. [max_time, batch_size, beam_width]. |
max_sequence_lengths | A Tensor типа int32. [batch_size]. |
end_token | A Tensor. Должен иметь тот же тип, что и step_ids. []. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
A Tensor. Имеет тот же тип, что и step_ids. [max_time, batch_size, beam_width]. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/seq2seq/gather_tree
