Fix the issue#4138 (tensorflow#4270)

Jingtian Peng · Vijay Vasudevan · commit 9374033eccc5 · 2016-09-09T21:29:39.000-07:00
diff --git a/tensorflow/models/rnn/translate/seq2seq_model.py b/tensorflow/models/rnn/translate/seq2seq_model.py
@@ -95,8 +95,8 @@ def __init__(self,
     softmax_loss_function = None
     # Sampled softmax only makes sense if we sample less than vocabulary size.
     if num_samples > 0 and num_samples < self.target_vocab_size:
-      w = tf.get_variable("proj_w", [size, self.target_vocab_size], dtype=dtype)
-      w_t = tf.transpose(w)
+      w_t = tf.get_variable("proj_w", [self.target_vocab_size, size], dtype=dtype)
+      w = tf.transpose(w_t)
       b = tf.get_variable("proj_b", [self.target_vocab_size], dtype=dtype)
       output_projection = (w, b)
 
