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使用的兩種方法小結就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持武林站長站。
新聞熱點
疑難解答