tf.contrib.rnn.static_state_saving_rnn
tf.nn.static_state_saving_rnn
tf.nn.static_state_saving_rnn( cell, inputs, state_saver, state_name, sequence_length=None, scope=None )
Defined in tensorflow/python/ops/rnn.py
.
See the guide: RNN and Cells (contrib) > Recurrent Neural Networks
RNN that accepts a state saver for time-truncated RNN calculation.
cell
: An instance of RNNCell
.inputs
: A length T list of inputs, each a Tensor
of shape [batch_size, input_size]
.state_saver
: A state saver object with methods state
and save_state
.state_name
: Python string or tuple of strings. The name to use with the state_saver. If the cell returns tuples of states (i.e., cell.state_size
is a tuple) then state_name
should be a tuple of strings having the same length as cell.state_size
. Otherwise it should be a single string.sequence_length
: (optional) An int32/int64 vector size [batch_size]. See the documentation for rnn() for more details about sequence_length.scope
: VariableScope for the created subgraph; defaults to "rnn".A pair (outputs, state) where: outputs is a length T list of outputs (one for each input) states is the final state
TypeError
: If cell
is not an instance of RNNCell.ValueError
: If inputs
is None
or an empty list, or if the arity and type of state_name
does not match that of cell.state_size
.
© 2018 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/api_docs/python/tf/nn/static_state_saving_rnn