简单的Tensorflow(3):使用feed_dict字典填充

来源:互联网 发布:淘宝运费险赔付多少钱 编辑:程序博客网 时间:2024/06/18 14:49

tensorflow还提供字典填充函数,使输入和输出更为简单:feed_dict = {}。例如:需要吧8和2填充到字典中,就需要占位符tensorflow.placeholder()而非变量,input1 = tf.placeholder(tf.float32),因为是一个元素不需要矩阵相乘,只要简单的乘法即可:tensorflow.multiply()


全部代码:

import tensorflow as tf#设置两个乘数,用占位符表示input1 = tf.placeholder(tf.float32)input2 = tf.placeholder(tf.float32)#设置乘积output = tf.multiply(input1, input2)with tf.Session() as sess:    #用feed_dict以字典的方式填充占位    print(sess.run(output, feed_dict={input1:[8.],input2:[2.]}))


结果是:

[ 16.]

原创粉丝点击