TensorFlow – 创建计算图

创建计算图

下面的代码在内存中创建一个默认计算图。

import tensorflow as tf

x = tf.Variable(3, name="x")

y = tf.Variable(4, name="y")

f = x*x*y + y + 2

代码中声明了变量和函数,这将会在内存中创建一个默认计算图。

默认计算图

默认计算图是TensorFlow默认生成的计算图。

>>> import tensorflow as tf
>>> x1 = tf.Variable(1)
>>> x1.graph is tf.get_default_graph( )
True

注意
当你在Python shell中进行试验时,通常会多次运行相同的命令,这可能会导致默认图中包含许多重复节点。例如,同个python shell中运行2次x = tf.variable (3, name="x"),会有2个x节点。
可以通过重启Python shell解决这个问题,但更方便的是通过运行tf.reset_default_graph()来重置默认计算图。

显式创建计算图

如果需要,可创建默认图之外的其他计算图。

新节点都会被添加到默认图中,如果想添加到新的图中,需要把新的图设置为默认图。

例如,下面代码中所示的变量x2的情况。

示例

>>> graph = tf.Graph( )
>>> with graph.as_default( ):
...     x2 = tf.Variable(2)
...
>>> x2.graph is graph
True
>>> x2.graph is tf.get_default_graph( )
False
>>>

如上所示,要使用新图,创建一个图并使用with块将其临时设置为默认图形。



浙ICP备17015664号 浙公网安备 33011002012336号 联系我们 网站地图  
@2019 qikegu.com 版权所有,禁止转载