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

举报
北山啦 发表于 2021/03/09 00:53:24 2021/03/09
【摘要】 在TensorFlow2.x中执行TensorFlow1.x代码的静态图执行模式 改为图执行模式 TensorFlow2虽然和TensorFlow1.x有较大差异,不能直接兼容。但实际上还是提供了对TensorFlow1.x的API支持 TensorFlow 2中执行或开发TensorFlow1.x代码,可以做如下处理: 导入TensorFlow时使用 i...

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

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


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

  1. 导入TensorFlow时使用
import tensorflow.compat.v1 as tf

  
 
  • 1
  1. 禁用即时执行模式
tf.disable_eager_execution()

  
 
  • 1

简单两步即可

举例

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)

  
 
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7

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

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

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

  
 
  • 1
  • 2
  • 3
  • 4
  • 5

在这里插入图片描述

文章来源: blog.csdn.net,作者:北山啦,版权归原作者所有,如需转载,请联系作者。

原文链接:blog.csdn.net/qq_45176548/article/details/114343184

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。