python - Eager Execution 与 tf.function 的关系

Question

我的代码分布在多个文件和函数中，因此在这里引用它是不切实际的。我想，如果我很好地理解了eager execution和tf.function，我可以自己解决。有人可以帮我解决以下问题吗？

@tf.function 装饰器仅在禁用急切执行时才起作用吗？

@tf.function 装饰器和 Eager 执行模式有什么关系？

TensorFlow 如何在内部切换 Eager 模式和非 Eager 模式？当我禁用 Eager Execution 并且我的代码中有一些错误时，我仍然会引用文件，这些文件向我建议 Eager Execution 可能正在内部发生。

Answer 1

让我试着解释一下。希望它会很有用。

@tf.function 装饰器仅在禁用急切执行时才起作用？
不，实际上 tf.function 是在启用 Eager 模式时加速执行的东西

@tf.function 装饰器和 Eager 执行模式有什么关系？
@tf.function 将导致 tensorflow 签名工作并加速其中的那些操作的执行。

TensorFlow 如何在内部切换 Eager 模式和非 Eager 模式？
tf.function 和 AutoGraph 通过生成代码并将其跟踪到 TensorFlow 图中来工作。所以当你第一次调用一个@tf.function 修饰的函数时，tensorflow 会先把它转换成一个图，然后执行它，之后，当你再次调用这个函数时，它只会执行这个图。

您还可以查看 tensorflow 文档。
https://www.tensorflow.org/guide/function