low efficiency of `tf.matmul`

[wizardforcel]

tf.matmul(tf.one_hot(self.u_i, depth=args.num_of_nodes), self.embedding)

Is it better to use tf.gather?

tf.gather(self.embedding, self.u_i)