Merge pull request tensorflow#1427 from kepingwang/master

stack outputs instead of concat outputs
tarasglek · May 1, 2017 · 3a3c5b9 · 3a3c5b9
2 parents 428212c + a9e9982
commit 3a3c5b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tutorials/rnn/ptb/ptb_word_lm.py b/tutorials/rnn/ptb/ptb_word_lm.py
@@ -157,7 +157,7 @@ def attn_cell():
         (cell_output, state) = cell(inputs[:, time_step, :], state)
         outputs.append(cell_output)
 
-    output = tf.reshape(tf.concat(axis=1, values=outputs), [-1, size])
+    output = tf.reshape(tf.stack(axis=1, values=outputs), [-1, size])
     softmax_w = tf.get_variable(
         "softmax_w", [size, vocab_size], dtype=data_type())
     softmax_b = tf.get_variable("softmax_b", [vocab_size], dtype=data_type())