TensorFlow 模型导出

请提供最小可复现的代码,原则上来说加@tf.function需要非常谨慎