tensorflow会话控制-【老鱼学tensorflow】

在tensorflow中,当定义好结构后,就要通过tf.session()来建立运行时的会话。
本例子应该不难理解,我们用tensorflow来计算一下一个1行2列的矩阵和2行1列矩阵的乘积:

import tensorflow as tf

# 1行2列的矩阵
mat1 = tf.constant([[3, 3]])
# 2行1列的矩阵
mat2 = tf.constant([[2],
                    [2]])

out = tf.matmul(mat1, mat2)

sess = tf.Session()
print(sess.run(out))
sess.close()

输出为:

[[12]]

当然,如果读者熟悉python中的with语句,则也可以用with语句来进行,这样就可以不用管sess.close()方法了,因为with会考虑到自动调用相应资源的释放函数

with tf.Session() as sess:
    print(sess.run(out))

这里顺便也用numpy的方式来计算一下相同的矩阵乘法:

import numpy as np
mat1 = np.array([[3, 3]])
mat2 = np.array([[2],
                 [2]])
print(np.dot(mat1, mat2))

输出也一样为:

[[12]]
posted @   dreampursuer  阅读(231)  评论(0编辑  收藏  举报
编辑推荐:
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
· AI与.NET技术实操系列(二):开始使用ML.NET
· 记一次.NET内存居高不下排查解决与启示
阅读排行:
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
点击右上角即可分享
微信分享提示