TensorFlow演示注入机制
一 实例
定义占位符,使用feed机制将具体数值(3和4)通过占位符传入,并进行相加和相乘运算。
二 代码
import tensorflow as tf a = tf.placeholder(tf.int16) b = tf.placeholder(tf.int16) add = tf.add(a, b) mul = tf.multiply(a, b) #a与b相乘 with tf.Session() as sess: # Run every operation with variable input print ("相加: %i" % sess.run(add, feed_dict={a: 3, b: 4})) print ("相乘: %i" % sess.run(mul, feed_dict={a: 3, b: 4}))
三 运行结果
相加: 7
相乘: 12
四 说明
使用注入机制,将具体的实参注入到相应的placeholder中。feed只在调用它的方法内有效,方法结束后feed就会消失。