tensorflow代码-初识-矩阵相乘

学习tensorflow之路 一：矩阵相乘

步骤：构建一个Graph，将需要计算的变量放入Graph中进行计算

一个简单的矩阵相乘案例：

"""
    创建一个Graph，然后启动
"""


#导入需要的模块
import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

#创建两个常量，m1,m2两个矩阵分别是 m1[1×2]一行两列 m2[2×1]两行一列
m1 = tf.constant([[3,3]])
m2 = tf.constant([[2],[3]])

#创建一个乘法操作，将m1，m2两个矩阵作为参数传入
product = tf.matmul(m1,m2)
print(product)#显示结果

#--------从此处开始tensorflow操作，前面的代码只是定义了代码，并没有执行------------------

#定义一个会话，启动默认Graph
sess = tf.Session()
#使用Session的run方法执行上面produce（即：矩阵乘法操作）
result = sess.run(product)
print(result)#显示结果
sess.close()#关闭会话

结果：

[[15]]

下面还有一种简单的创建会话的方式，即使用with的方法创建会话，这样创建会话省去了关闭会话的操作

#定于会话方法2：此种方法是一种更加简单的方式，省去了关闭会话的操作
with tf.Session() as sess:
    result = sess.run(product)
    print(result)