在什么情况下可以使用@tf.function?



我听说tf.function可以构建图形并加速计算。但是我们什么时候可以使用它呢?

从另一个答案来看,似乎是:

当我们不使用内置函数

有更具体的答案吗?

这是一个统计tf.function优点的链接。假设你有一个损失函数,在函数上添加这个关键字可以帮助Tensorflow更快地运行它,但是你可能会遇到像

这样的错误。
ValueError: tf.function-decorated function tried to create variables on non-first call.

如果在第一次运行时没有初始化所有变量。

相关内容

  • 没有找到相关文章

最新更新