`x_emb = tf.nn.embedding_lookup(self.embeddings, self.x)` The dimension[1] should be 1014. 