diff options
author | 2018-03-14 21:17:39 -0400 | |
---|---|---|
committer | 2018-03-14 18:17:39 -0700 | |
commit | 96ab843c3af177cc0cf7f9ec2caa3d95369d9c4c (patch) | |
tree | d1350e3de6424f2fd2918c4b09b2761839267b0b /tensorflow/contrib/rnn/python/ops/lstm_ops.py | |
parent | e605245fbd6ef370ac9a07dddfa733da4705017f (diff) |
updating documentation (#17406)
Diffstat (limited to 'tensorflow/contrib/rnn/python/ops/lstm_ops.py')
-rw-r--r-- | tensorflow/contrib/rnn/python/ops/lstm_ops.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/contrib/rnn/python/ops/lstm_ops.py b/tensorflow/contrib/rnn/python/ops/lstm_ops.py index 4eb4fbcd92..9e61fc54d1 100644 --- a/tensorflow/contrib/rnn/python/ops/lstm_ops.py +++ b/tensorflow/contrib/rnn/python/ops/lstm_ops.py @@ -480,8 +480,7 @@ class LSTMBlockWrapper(base_layer.Layer): """Run this LSTM on inputs, starting from the given state. Args: - inputs: `3-D` tensor with shape `[time_len, batch_size, input_size]` - or a list of `time_len` tensors of shape `[batch_size, input_size]`. + inputs: `3-D` tensor with shape `[time_len, batch_size, input_size]`. initial_state: a tuple `(initial_cell_state, initial_output)` with tensors of shape `[batch_size, self._num_units]`. If this is not provided, the cell is expected to create a zero initial state of type `dtype`. |