Commit e3ad49a5 authored by Andrew Selle's avatar Andrew Selle Committed by GitHub
Browse files

Merge pull request #1017 from aselle/concat_v2

Update concat_v2 to be concat to match 1.0 final
parents 596c9e23 63447939
......@@ -100,7 +100,7 @@ def build_input(dataset, data_path, batch_size, mode):
labels = tf.reshape(labels, [batch_size, 1])
indices = tf.reshape(tf.range(0, batch_size, 1), [batch_size, 1])
labels = tf.sparse_to_dense(
tf.concat_v2(values=[indices, labels], axis=1),
tf.concat(values=[indices, labels], axis=1),
[batch_size, num_classes], 1.0, 0.0)
assert len(images.get_shape()) == 4
......
......@@ -146,7 +146,7 @@ class PTBModel(object):
(cell_output, state) = cell(inputs[:, time_step, :], state)
outputs.append(cell_output)
output = tf.reshape(tf.concat_v2(outputs, 1), [-1, size])
output = tf.reshape(tf.concat(outputs, 1), [-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())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment