tf.gather和tf.gather_nd、tf.cast、tf.greater