tf.contrib.legacy_seq2seq.basic_rnn_seq2seq
Basic RNN sequence-to-sequence model.
tf.contrib.legacy_seq2seq.basic_rnn_seq2seq( encoder_inputs, decoder_inputs, cell, dtype=tf.dtypes.float32, scope=None )
This model first runs an RNN to encode encoder_inputs into a state vector, then runs decoder, initialized with the last encoder state, on decoder_inputs. Encoder and decoder use the same RNN cell type, but don't share parameters.
Args | |
---|---|
encoder_inputs | A list of 2D Tensors [batch_size x input_size]. |
decoder_inputs | A list of 2D Tensors [batch_size x input_size]. |
cell | tf.compat.v1.nn.rnn_cell.RNNCell defining the cell function and size. |
dtype | The dtype of the initial state of the RNN cell (default: tf.float32). |
scope | VariableScope for the created subgraph; default: "basic_rnn_seq2seq". |
Returns | |
---|---|
A tuple of the form (outputs, state), where: outputs: A list of the same length as decoder_inputs of 2D Tensors with shape [batch_size x output_size] containing the generated outputs. state: The state of each decoder cell in the final time-step. It is a 2D Tensor of shape [batch_size x cell.state_size]. |
© 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/basic_rnn_seq2seq