用户输入两个值,计算两个值的乘积
import tensorflow as tf
#placeholder 类似于c++的cin,要求用户运行时输入
input1 = tf.placeholder(tf.float32, [2, 2])
input2 = tf.placeholder(tf.float32, [2, 2])
input3 = tf.placeholder(tf.float32)
input4 = tf.placeholder(tf.float32)
#矩阵乘法
output1 = tf.matmul(input1, input2)
#数字乘法
output2 = tf.multiply(input3, input4)
with tf.Session() as sess:
#feed_dict是关键字,必须写成这个名字,后面紧跟传入的参数
print (sess.run(output1, feed_dict ={input1:[[1, 2], [3, 4]], input2:[[1, 1], [1, 1]]}))
print (sess.run(output2, feed_dict = {input3:3, input4:4}))
print (sess.run(output2, feed_dict = {input3:[4], input4:[5]}))
[[ 3. 3.]
[ 7. 7.]]
12.0
[ 20.]