我听说tf.function
可以构建图形并加速计算。但是我们什么时候可以使用它呢?
从另一个答案来看,似乎是:
当我们不使用内置函数
时
有更具体的答案吗?
这是一个统计tf.function优点的链接。假设你有一个损失函数,在函数上添加这个关键字可以帮助Tensorflow更快地运行它,但是你可能会遇到像
这样的错误。ValueError: tf.function-decorated function tried to create variables on non-first call.
如果在第一次运行时没有初始化所有变量。