您的位置:首页 > Web前端

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

2017-11-29 09:21 363 查看
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.]
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: