Forgotten squeeze

Author: metric-space

site/en/tutorials/text/word2vec.ipynb

@@ -750,7 +750,7 @@
         "      negative_sampling_candidates = tf.expand_dims(\n",
         "          negative_sampling_candidates, 1)\n",
         "\n",
-        "      context = tf.concat([context_class, negative_sampling_candidates], 0)\n",
+        "      context = tf.concat([tf.squeeze(context_class,1), negative_sampling_candidates], 0)\n",
         "      label = tf.constant([1] + [0]*num_ns, dtype=\"int64\")\n",
         "\n",
         "      # Append each element from the training example to global lists.\n",