TensorFlow 2 tf.function decorator

来源：https://stackoverflow.com/questions/60059410/tensorflow-2-tf-function-decorator