7.4.4 利用tf.function加快运行速度