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 成為默認圖。