在TensorFlow2.x中执行TensorFlow1.x代码的静态图执行模式

改为图执行模式
在这里插入图片描述
TensorFlow2虽然和TensorFlow1.x有较大差异,不能直接兼容。但实际上还是提供了对TensorFlow1.x的API支持


TensorFlow 2中执行或开发TensorFlow1.x代码,可以做如下处理:

  1. 导入TensorFlow时使用
import tensorflow.compat.v1 as tf
  1. 禁用即时执行模式
tf.disable_eager_execution()

简单两步即可

举例

import tensorflow.compat.v1 as tf
tf.disable_eager_execution()

node1 = tf.constant(3.0)
node2 = tf.constant(4.0)
node3 = tf.add(node1,node2)
print(node3)

由于是图执行模式,这时仅仅是建立了计算图,但没有执行

定义好计算图后,需要建立一个Session,使用会话对象来实现执行图的执行

sess = tf.Session()
print("node1:",sess.run(node1))
print("node2:",sess.run(node2))
print("node3:",sess.run(node3))
Session.close()

在这里插入图片描述

在这里插入图片描述

GitHub 加速计划 / te / tensorflow
184.55 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:2 个月前 )
a49e66f2 PiperOrigin-RevId: 663726708 3 个月前
91dac11a This test overrides disabled_backends, dropping the default value in the process. PiperOrigin-RevId: 663711155 3 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐