是的，理论上是tf.function这样的。这使用了一个名为 AutoGraph 的 tensorflow 模块，基本上将您的 python/numpy 操作转换为 tensorflow 操作。

然而，并非所有 python 操作都可以转换。诸如打印、附加到列表和改变全局变量之类的事情在图形模式下不起作用！

我建议你看一下官方指南，看看它是如何使用的，以及它是否适用于你的情况。