在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