Spec-Zone.ru › TensorFlow 1.15

tf.contrib.legacy_seq2seq.tied_rnn_seq2seq

Модель последовательности «вопрос-ответ» с RNN, использующей связанные параметры кодера и декодера.

tf.contrib.legacy_seq2seq.tied_rnn_seq2seq(
    encoder_inputs, decoder_inputs, cell, loop_function=None,
    dtype=tf.dtypes.float32, scope=None
)

Данная модель сначала запускает RNN для кодирования входных данных кодера в вектор состояния, а затем запускает декодер, инициализированный последним состоянием кодера, на входных данных декодера. Кодер и декодер используют одну и ту же ячейку RNN и совместно используют параметры.

Аргументы
encoder_inputs Список 2D тензоров [размер_пакета x размер_входа].
decoder_inputs Список 2D тензоров [размер_пакета x размер_входа].
cell tf.compat.v1.nn.rnn_cell.RNNCell, определяющий функцию и размер ячейки.
loop_function Если не равно None, эта функция будет применена к i-му выводу, чтобы сгенерировать i+1-й вход, и входные данные декодера будут проигнорированы, за исключением первого элемента («символ GO»). Подробнее см. rnn_decoder.
dtype Тип данных начального состояния ячейки RNN (по умолчанию: tf.float32).
scope Область переменных для созданной подграфа; по умолчанию: "tied_rnn_seq2seq".
Возвращаемое значение
Кортеж вида (выходы, состояние), где: выходы: Список той же длины, что и decoder_inputs, содержащий 2D тензоры с формой [размер_пакета x размер_выхода], содержащие сгенерированные выходы. состояние: Состояние каждой ячейки декодера в каждом шаге во времени. Это список длиной len(decoder_inputs) — по одному элементу для каждого шага во времени. Это 2D тензор формы [размер_пакета x размер_состояния_ячейки].

© 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/legacy_seq2seq/tied_rnn_seq2seq

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API