在TensorFlow中是通過session進行交互的,使用session有兩種方法。下面通過一個簡單的例子(兩個矩陣相乘)說一下
{[3,1] 與{[5,2] 相乘
1
|
[ 1 , 2 ]} [ 2 , 4 ]} |
代碼
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
#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使用的兩種方法小結就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持服務器之家。
原文鏈接:https://blog.csdn.net/qq_33374476/article/details/76728323