欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

2-1 Tensorflow --创建图、启动图

程序员文章站 2022-06-20 08:53:56
创建图、启动图import tensorflow as tf# 创建一个常量op (1, 2)m1 = tf.constant([[3, 3]])# 创建一个常量op (2, 1)m2 = tf.constant([[2], [3]])# 创建一个矩阵乘法op,将m1和m2传入product = tf.matmul(m1, m2)print(product)'''Tensor("MatMul:0", shape=(1, 1), dtype=int32)常理应该是15,但是.....

2-1 Tensorflow --创建图、启动图
2-1 Tensorflow --创建图、启动图
创建图、启动图

import tensorflow as tf

# 创建一个常量op   (1, 2)
m1 = tf.constant([[3, 3]])
# 创建一个常量op   (2, 1)
m2 = tf.constant([[2], [3]])
# 创建一个矩阵乘法op,将m1和m2传入
product = tf.matmul(m1, m2)
print(product)
'''
Tensor("MatMul:0", shape=(1, 1), dtype=int32)
常理应该是15,但是结果是一个Tensor
因为我们上述只是定义了一些op,这么op并没有放到会话和图中,只是定义了这些op,并没有执行
'''

# 1.定义一个会话,会有默认的图
sess = tf.Session()
# 调用sess的run方法执行上述的乘法op,触发了图中的三个op
result = sess.run(product)
print(result)
sess.close()

# 2.上述会话定义还要关闭,一般用with
with tf.Session() as sess:
    result = sess.run(product)
    print(result)

本文地址:https://blog.csdn.net/qq_42363032/article/details/107356645