aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/rnn/python/ops/lstm_ops.py
diff options
context:
space:
mode:
authorGravatar Russell Klopfer <russell@klopfer.us>2018-03-14 21:17:39 -0400
committerGravatar Frank Chen <frankchn@gmail.com>2018-03-14 18:17:39 -0700
commit96ab843c3af177cc0cf7f9ec2caa3d95369d9c4c (patch)
treed1350e3de6424f2fd2918c4b09b2761839267b0b /tensorflow/contrib/rnn/python/ops/lstm_ops.py
parente605245fbd6ef370ac9a07dddfa733da4705017f (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.py3
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`.