TensorFlow Session使用的两种方法小结

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使用的两种方法小结就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

以上是 TensorFlow Session使用的两种方法小结 的全部内容, 来源链接: utcz.com/z/319028.html

回到顶部