Tensorflow Tf.gather