为何调用as_default().__enter__()无法设置TensorFlow默认会话？

这是个非常好的问题，核心在于TensorFlow的as_default()返回的上下文管理器的生命周期管理，下面我来拆解细节：

第一段代码失效的原因 #

先看你的第一段代码：

import tensorflow as tf
x = tf.Session()
x.as_default().__enter__()
print(tf.get_default_session()) # 输出"None"

问题出在这几个步骤：

1. 调用x.as_default()时，会生成一个临时的上下文管理器对象，这个对象的职责就是管理默认会话的“进入”和“退出”逻辑。
2. 你直接调用它的__enter__()方法，确实会把x设为当前默认会话，但因为这个上下文管理器对象没有被任何变量保存（相当于一个匿名对象），Python的垃圾回收机制会立刻把它回收掉。
3. 当上下文管理器被回收时，Python会自动调用它的__exit__()方法——这个方法的作用就是恢复之前的默认会话状态（也就是移除刚刚设置的x作为默认会话）。
4. 等你执行print语句时，默认会话已经被清除了，所以输出None。

两段代码的核心差异 #

再看第二段用with块的代码：

import tensorflow as tf
with tf.Session().as_default():
    print(tf.get_default_session()) # 显示会话对象

和第一段的关键区别在于上下文管理器的生命周期：

• with语句会自动持有as_default()返回的上下文管理器对象，直到with块内的所有代码执行完毕，才会调用__exit__()方法。所以在with块内部，默认会话的设置一直有效，你能正常获取到会话对象。
• 手动调用__enter__()的方式绕过了with的生命周期管理，导致上下文管理器被提前回收，默认会话的设置也随之失效。

另外还有个细节：with语句是Python管理上下文资源的标准方式，它会自动处理异常场景——哪怕with块内抛出错误，__exit__()也会被执行，确保默认会话的状态被正确恢复，避免资源泄漏或状态混乱。而手动调用__enter__()和__exit__()很容易遗漏异常处理，导致状态异常。

手动实现类似with的效果（不推荐） #

如果非要手动模拟with的行为，你需要把上下文管理器保存下来，避免被提前回收：

import tensorflow as tf
x = tf.Session()
# 保存上下文管理器对象
ctx = x.as_default()
ctx.__enter__()
print(tf.get_default_session()) # 现在会输出会话对象
# 用完后必须手动调用__exit__()，恢复状态
ctx.__exit__(None, None, None)

不过这种方式非常不推荐，还是建议用官方推荐的with块来管理默认会话，避免踩坑。

内容的提问来源于stack exchange，提问作者Eric Auld