feed_dict使用

TF在程序的编写上,一般是先初始化,给变量用tf.placeholder函数写占位符,类似于函数参数,运行时必须传入具体的数值。

那么传值的责任就落在了feed_dict函数上,具体的用法参考下面小程序。

# -*- coding: utf-8 -*-
"""
Created on Wed Jun 13 10:08:00 2018

@author: Administrator
"""

import tensorflow as tf

# 定义placeholder
input1 = tf.placeholder(tf.float32)
input2 = tf.placeholder(tf.float32)

# 定义乘法运算
#因为是一个数值相乘,只要简单的乘法即可:tf.multiply()
output = tf.multiply(input1, input2) 

# 通过session执行乘法运行
with tf.Session() as sess:
    # 执行时要传入placeholder的值
    print(sess.run(output, feed_dict = {input1:[2.], input2: [3.]}))

猜你喜欢

转载自blog.csdn.net/qq_27871973/article/details/80676069