tensorflow的基本用法(二)——Session

文章作者: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]]
    原文作者:SnailTyan
    原文地址: https://www.jianshu.com/p/bd2983ade316
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞