---
title: 添加@tf.function装饰器
description: "通常，官方发布或设计良好的脚本都会自带@tf.function装饰器或者允许您通过脚本入参控制，您无需额外添加。"
url: https://www.hiascend.com/document/detail/zh/TensorFlowCommercial/latest/migration/tfmigr2/tfmigr2_000014.html
sourcePath: /source/zh/TensorFlowCommercial/900/migration/tfmigr2/tfmigr2_000014.html
indexId: 0194c642462e114309bba3f48377665136062bba168d07375828dce864f3e8c373
---
# 添加@tf.function装饰器

通常，官方发布或设计良好的脚本都会自带@tf.function装饰器或者允许您通过脚本入参控制，您无需额外添加。

特别地，使用Keras下Model.fit接口训练的脚本，TF2会在接口内部封装function，同样无需额外添加。

如果脚本中没有添加@tf.function装饰器，您可以先阅读TF2对@tf.function的使用说明，详情见链接(https://www.tensorflow.org/api_docs/python/tf/function)，然后为您的训练/验证/推理函数添加@tf.function装饰器，并保证其在CPU或GPU环境下能正常工作。