代码:
import tensorflow as tf
#创建常量op
m1 = tf.constant([[3, 3]])
m2 = tf.constant([[2], [3]])
#创建一个矩阵乘法op, 把m1和m2传入
product = tf.matmul(m1, m2)
#只是定义了op,并没有执行,只是得到了一个tensor
print(product)
运行结果:
Tensor("MatMul:0", shape=(1, 1), dtype=int32)
代码:
#定义一个会话,启动默认的图
sess = tf.Session()
#调用sess的run方法来执行矩阵乘法op
#run(product)触发了图中3个op
result = sess.run(product)
print(result)
sess.close()
运行结果:
[[15]]
代码:
#启动图的另一个方法
#不需要执行关闭操作,语句执行结束,会自动关闭
with tf.Session() as sess:
#调用sess的run方法来执行矩阵乘法op
#run(product)触发了图中3个op
result = sess.run(product)
print(result)
运行结果:
[[15]]