Tensorflow—创建图

代码:

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]]

猜你喜欢

转载自blog.csdn.net/wangsiji_buaa/article/details/80200580