diff options
| author |  Yanbo Liang <ybliang8@gmail.com> | 2018-04-11 18:42:50 -0700 |
|---|---|---|
| committer |  Jonathan Hseu <vomjom@vomjom.net> | 2018-04-11 18:42:50 -0700 |
| commit | 079539b2e7acb1813cbfcdd2ab39f7bb77bc0467 (patch) | |
| tree | 1911637e69f0b015fd94bce7dc7d65163e5f0e32 | |
| parent | 1caeb2086e7e9d7e3cb85883f0af316cddcf1285 (diff) | |
Correct argument doc for BasicLSTMCell.call (#16554)
* Correct argument doc for BasicLSTMCell.call
* change self._num_units to num_units.
| -rw-r--r-- | tensorflow/python/ops/rnn_cell_impl.py | 4 |
1 files 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 |