aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/docs_src
diff options
context:
space:
mode:
authorGravatar Dustin Tran <trandustin@google.com>2018-08-13 17:59:45 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-08-13 18:03:57 -0700
commitab755c94beffc0531ca24b157858b3a10d10f397 (patch)
tree99c84b85d49668bd7c2e066d89d54ae419d03b38 /tensorflow/docs_src
parent64acb73e0a970a8f177854a4286979acd40903d6 (diff)
Fix recurrent tutorial's code snippet.
PiperOrigin-RevId: 208575390
Diffstat (limited to 'tensorflow/docs_src')
-rw-r--r--tensorflow/docs_src/tutorials/sequences/recurrent.md6
1 files changed, 2 insertions, 4 deletions
diff --git a/tensorflow/docs_src/tutorials/sequences/recurrent.md b/tensorflow/docs_src/tutorials/sequences/recurrent.md
index 715cc7856a..10d60f7966 100644
--- a/tensorflow/docs_src/tutorials/sequences/recurrent.md
+++ b/tensorflow/docs_src/tutorials/sequences/recurrent.md
@@ -77,9 +77,7 @@ The basic pseudocode is as follows:
words_in_dataset = tf.placeholder(tf.float32, [time_steps, batch_size, num_features])
lstm = tf.contrib.rnn.BasicLSTMCell(lstm_size)
# Initial state of the LSTM memory.
-hidden_state = tf.zeros([batch_size, lstm.state_size])
-current_state = tf.zeros([batch_size, lstm.state_size])
-state = hidden_state, current_state
+state = lstm.zero_state(batch_size, dtype=tf.float32)
probabilities = []
loss = 0.0
for current_batch_of_words in words_in_dataset:
@@ -112,7 +110,7 @@ words = tf.placeholder(tf.int32, [batch_size, num_steps])
lstm = tf.contrib.rnn.BasicLSTMCell(lstm_size)
# Initial state of the LSTM memory.
-initial_state = state = tf.zeros([batch_size, lstm.state_size])
+initial_state = state = lstm.zero_state(batch_size, dtype=tf.float32)
for i in range(num_steps):
# The value of state is updated after processing each batch of words.