【tensorflow】常量和变量的定义

源式羽语 2020-08-09

  • tensorflow 2.0 无法兼容 1.0 版本,所以需要加上一句 tf.compat.v1.disable_eager_execution(),以保障程序的正常运行。
  • 在 tensorflow 2.0 中,初始化变量和声明 Session 会话的函数均被放在了 tensorflow.compat.v1 中,调用相关函数时,需要加上这些前缀。

 

1.声明并打印常量和变量

代码:

import tensorflow as tf
tf.compat.v1.disable_eager_execution()              # 保证sess.run()能够正常运行
data1 = tf.constant(2, dtype=tf.int32)              # 声明常量
data2 = tf.Variable(10, name="var")                 # 声明变量
print(data1, data2)                                 # 打印数据的描述信息
init = tf.compat.v1.global_variables_initializer()  # 初始化变量
with tf.compat.v1.Session() as sess:                # 使用Session会话执行各种操作
    sess.run(init)
    print(sess.run(data1), sess.run(data2))

输出结果:

【tensorflow】常量和变量的定义

 

相关推荐