TensorFlow Session使用的两种方法 TensorFlow Session使用的两种方法小结

软件发布|下载排行|最新软件

当前位置:首页IT学院IT技术

TensorFlow Session使用的两种方法 TensorFlow Session使用的两种方法小结

星璇love   2021-03-31 我要评论

TensorFlow Session

在TensorFlow中是通过session进行交互的,使用session有两种方法。下面通过一个简单的例子(两个矩阵相乘)说一下

{[3,1] 与{[5,2] 相乘

[1,2]} [2,4]}

代码

#encoding=utf-8
import tensorflow as tf

matrix1 = tf.constant([[3,1],[1,2]])
matrix2 = tf.constant([[5,2],[2,4]])

product = tf.matmul(matrix1,matrix2) #矩阵乘法

#session method 1
sess = tf.Session()
result = sess.run(product)
print(result)
sess.close()

#session method 2
with tf.Session() as sess2:
 result2 = sess2.run(product)
 print(result2)

结果

以上这篇TensorFlow Session使用的两种方法小结就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

猜您喜欢

Copyright 2022 版权所有 软件发布 访问手机版

声明:所有软件和文章来自软件开发商或者作者 如有异议 请与本站联系 联系我们