tensorflow的基本用法(二)

文章作者：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]]

参考资料

1. https://www.youtube.com/user/MorvanZhou