Tensorflow 学习1:图的创建和运行
从今天开始在B站上学习tensorflow,把自己的所学写在这里,方便以后回来温故。(课程链接:https://www.bilibili.com/video/av20542427/?p=4)
第一节:Tensorflow中图的创建和运行
图是由tensor,op来创建:
创建一个图:
首先我们要载入tensorflow包:
#载入tensorflow
import tensorflow as tf
而后我们可以创建图所需要的op并定义矩阵乘积op:
#创建向量op
m1 = tf.constant([[2,3]])
m2 = tf.constant([[3],[3]])
#创建矩阵乘积op,计算m1和m2的乘积
product = tf.matmul(m1,m2)
print(product)
这里有一些小细节值得新手注意:
- jupyter notebook的cell中的函数索引功能,有时候我们几天敲没代码就容易忘记方法的全称,就会给编程带来很大困扰。但是在jupyter notebook中,我们可以用先敲出tf.con然后单击键盘上的Tab键,便可以看到所有前缀是con的方法,进而选取我们要的tf.constant()
- 另一个细节是,print()出来的product并不是矩阵乘积的值,而是一个Tensor。这是因为,我们只是创建了一个op而并没有在图中运行它。
创建一个会话,并启动图:
首先我们要创建一个会话,然后在会话中运行上面的三个op:
#创建一个会话
sess = tf.Session()
#使用sess中的run方法来运行上面三个op
result = sess.run(product)
print(product)
sess.close()
每次启动会话之后,要记得关闭,为了方便起见,我们可以使用with函数来写这段代码:
with tf.session() as sess:
#使用sess中的run方法来运行上面三个op
result = sess.run(product)
print(product)
以上便是第一讲内容。