我运行的代码跟你的一样，可是在最后进行测试的时候发现我运行的结果是普通语法块的运行速度比tensorflow运行的要快
来源：3-7 tf.function函数转换
Jony_Li
2019-08-05
是我的电脑tensorflow配置出问题了吗，我的电脑cpu是八代i7,gpu是1060，安装的是TensorFlow2.0
我的环境是这样的
安装的tensorflow版本是这样的
cuna是这样的
我前面的运行结果跟你的一样，
但是我最后的运行结果却是这样
请问是哪里出了问题
还有就是，如果我是直接在定义的函数前面@tf.function后他的结果又对了
写回答
1回答
-
正十七
2019-08-18
对于你的最后两个例子，我理解应该是加了一次@tf.function，已经做了优化，所以第二次也对了。如果调换顺序然后重新执行，应该还是错误的结果。
看样子像是只有@tf.function做了优化而tf.function()没有。我得去检查一下底层实现看是不是这样哈。
012019-11-29
相似问题