diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/ops/embeddings_ops.py')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/ops/embeddings_ops.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/learn/python/learn/ops/embeddings_ops.py b/tensorflow/contrib/learn/python/learn/ops/embeddings_ops.py index 5ac9bfd808..fa3b7323e3 100644 --- a/tensorflow/contrib/learn/python/learn/ops/embeddings_ops.py +++ b/tensorflow/contrib/learn/python/learn/ops/embeddings_ops.py @@ -59,7 +59,7 @@ def embedding_lookup(params, ids, name='embedding_lookup'): ids_flat = array_ops_.reshape( ids, math_ops.reduce_prod(shape, keep_dims=True)) embeds_flat = nn.embedding_lookup(params, ids_flat, name) - embed_shape = array_ops_.concat_v2([shape, [-1]], 0) + embed_shape = array_ops_.concat([shape, [-1]], 0) embeds = array_ops_.reshape(embeds_flat, embed_shape) embeds.set_shape(ids.get_shape().concatenate(params.get_shape()[1:])) return embeds |