tf1.py 747 B

1234567891011121314151617181920
  1. import tensorflow as tf
  2. assert tf.__version__.startswith('1.')
  3. # 1.创建计算图阶段
  4. # 创建2个输入端子,指定类型和名字
  5. a_ph = tf.placeholder(tf.float32, name='variable_a')
  6. b_ph = tf.placeholder(tf.float32, name='variable_b')
  7. # 创建输出端子的运算操作,并命名
  8. c_op = tf.add(a_ph, b_ph, name='variable_c')
  9. # 2.运行计算图阶段
  10. # 创建运行环境
  11. sess = tf.InteractiveSession()
  12. # 初始化操作也需要作为操作运行
  13. init = tf.global_variables_initializer()
  14. sess.run(init) # 运行初始化操作,完成初始化
  15. # 运行输出端子,需要给输入端子赋值
  16. c_numpy = sess.run(c_op, feed_dict={a_ph: 2., b_ph: 4.})
  17. # 运算完输出端子才能得到数值类型的c_numpy
  18. print('a+b=',c_numpy)