tensorflow的基本用法(二)

版权声明:博客文章都是作者辛苦整理的,转载请注明出处,谢谢! https://blog.csdn.net/Quincuntial/article/details/69789285

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

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券