Python tf.Graph.as_default用法及代码示例

用法

as_default()

返回使此 Graph 成为默认图的上下文管理器。

如果要在同一进程中创建多个图形，则应使用此方法。为方便起见，提供了一个全局默认图，如果您不显式创建新图，所有操作都将添加到该图中。

将此方法与 with 关键字一起使用以指定应将在块范围内创建的操作添加到此图中。在这种情况下，一旦退出with 的范围，之前的默认图再次设置为默认值。有一个堆栈，因此可以有多个嵌套级别的as_default 调用。

默认图是当前线程的属性。如果您创建一个新线程，并希望在该线程中使用默认图形，则必须在该线程的函数中显式添加 with g.as_default():。

以下代码示例是等效的：

# 1. Using Graph.as_default():
g = tf.Graph()
with g.as_default():
  c = tf.constant(5.0)
  assert c.graph is g

# 2. Constructing and making default:
with tf.Graph().as_default() as g:
  c = tf.constant(5.0)
  assert c.graph is g

如果启用了即刻执行，则在此上下文管理器下创建的操作将被添加到图中，而不是即刻执行。