1. 程式人生 > >Tensorflow fetch和feed

Tensorflow fetch和feed

dict 數據 lac run NPU session input hold ESS

import tensorflow as tf

#Fetch
input1 = tf.constant(1.0)
input2 = tf.constant(3.0)
input3 = tf.constant(5.0)

add = tf.add(input2,input3)
mul = tf.multiply(input1,add)

with tf.Session() as sess:
result = sess.run([mul,add])
print(result)

############輸出 [8.0, 8.0]

#Feed
#創建占位符
input4 = tf.placeholder(tf.float32)
input5 = tf.placeholder(tf.float32)
output = tf.multiply(input4,input5)

with tf.Session() as sess:
#Feed的數據以字典的形式傳入
print(sess.run(output,feed_dict={input4:[7.0],input5:[9.0]}))

#輸出 [ 63.]

Tensorflow fetch和feed