@tf.function 是不是只能接受tf 的计算？我 train_loss.result()操作的代码在加了@tf.function 后就会无法执行

首页课程实战体系课手记专栏慕课教程

@tf.function 是不是只能接受tf 的计算？我 train_loss.result()操作的代码在加了@tf.function 后就会无法执行

来源：3-8 @tf.function函数转换

weixin_慕村2214139

2020-06-19

图片描述

写回答

1回答

正十七

正十七

2020-07-30

同学，你好，你的理解是对的，用了tf.function的标注就只能使用tensorflow的操作。因为它会把整个函数优化成tensorflow的图，对于其他操作它无法做出优化。

0

0

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

相似问题

什么时候用@tf.function，以及与tf.py_function的区别

回答 1

@tf.function自定义train_step训练速度缓慢

回答 1

我运行的代码跟你的一样，可是在最后进行测试的时候发现我运行的结果是普通语法块的运行速度比tensorflow运行的要快

回答 1

请教：去掉@tf.function而言，converge_to_2和add_21函数的输出类型不同？

回答 1

display_tf_code(converge_to_2)并不行

回答 1

打开慕课网App查看更多内容