TensorFlow Session使用的两种方法小结

技术  /  管理员 发布于 7年前   166

TensorFlow Session

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

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

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

代码

#encoding=utf-8import tensorflow as tfmatrix1 = tf.constant([[3,1],[1,2]])matrix2 = tf.constant([[5,2],[2,4]])product = tf.matmul(matrix1,matrix2) #矩阵乘法#session method 1sess = tf.Session()result = sess.run(product)print(result)sess.close()#session method 2with tf.Session() as sess2: result2 = sess2.run(product) print(result2)

结果

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