From 079539b2e7acb1813cbfcdd2ab39f7bb77bc0467 Mon Sep 17 00:00:00 2001 From: Yanbo Liang Date: Wed, 11 Apr 2018 18:42:50 -0700 Subject: Correct argument doc for BasicLSTMCell.call (#16554) * Correct argument doc for BasicLSTMCell.call * change self._num_units to num_units. --- tensorflow/python/ops/rnn_cell_impl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/ops/rnn_cell_impl.py b/tensorflow/python/ops/rnn_cell_impl.py index cbc2dcf419..54f4e0f240 100644 --- a/tensorflow/python/ops/rnn_cell_impl.py +++ b/tensorflow/python/ops/rnn_cell_impl.py @@ -599,9 +599,9 @@ class BasicLSTMCell(LayerRNNCell): Args: inputs: `2-D` tensor with shape `[batch_size, input_size]`. state: An `LSTMStateTuple` of state tensors, each shaped - `[batch_size, self.state_size]`, if `state_is_tuple` has been set to + `[batch_size, num_units]`, if `state_is_tuple` has been set to `True`. Otherwise, a `Tensor` shaped - `[batch_size, 2 * self.state_size]`. + `[batch_size, 2 * num_units]`. Returns: A pair containing the new hidden state, and the new state (either a -- cgit v1.2.3