文章作者:Tyan
博客:noahsnail.com | CSDN | 简书
本文主要是使用tensorflow进行矩阵的乘法运算。代码中介绍了两种不同的使用session的方式。Demo源码及解释如下:
#!/usr/bin/env python
# _*_ coding: utf-8 _*_
import tensorflow as tf
import numpy as np
# 定义两个矩阵
matrix1 = tf.constant([[3, 3]])
matrix2 = tf.constant([[2], [2]])
# 定义矩阵乘法
product = tf.matmul(matrix1, matrix2)
# 运行矩阵乘法,session用法一
sess = tf.Session()
result = sess.run(product)
print 'Session用法一'
print result
sess.close()
## session用法二,不用考虑close,会自动关闭
with tf.Session() as sess:
result = sess.run(product)
print 'Session用法二'
print result
执行结果如下图:
Session用法一
[[12]]
Session用法二
[[12]]