Python tf.compat.v1.Session.as_default用法及代码示例

用法

as_default()

返回使该对象成为默认会话的上下文管理器。

与 with 关键字一起使用以指定应在此会话中执行对 tf.Operation.run 或 tf.Tensor.eval 的调用。

c = tf.constant(..)
sess = tf.compat.v1.Session()

with sess.as_default():
  assert tf.compat.v1.get_default_session() is sess
  print(c.eval())

要获取当前默认会话，请使用 tf.compat.v1.get_default_session 。

注意： as_default上下文管理器才不是退出上下文时关闭会话，并且必须显式关闭会话。

c = tf.constant(...)
sess = tf.compat.v1.Session()
with sess.as_default():
  print(c.eval())
# ...
with sess.as_default():
  print(c.eval())

sess.close()

或者，您可以使用with tf.compat.v1.Session(): 创建一个在退出上下文时自动关闭的会话，包括在引发未捕获的异常时。

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

注意：输入 with sess.as_default(): 块不会影响当前的默认图形。如果您使用多个图，并且 sess.graph 与 tf.compat.v1.get_default_graph 的值不同，则必须显式输入 with sess.graph.as_default(): 块以使 sess.graph 成为默认图。