无法启用 Tensorflows Eager execution

Question

我有一个安装了 Tensorflow 2.0.0-beta1 的 conda 环境。但是，每当我导入 tensorflow 并尝试启用急切执行时，我都会收到错误消息：

AttributeError: module 'tensorflow' has no attribute 'enable_eager_execution'

我唯一的运行代码是：

import tensorflow as tf
print(tf.__version__)
tf.enable_eager_execution()

这是 tensorflow 2.0 beta 模块的错误还是我的安装有问题？

Answer 1

在 ternsorflow 2.0 中，enable_eager_execution 方法被移动到 tf.compat.v1 模块。以下工作在 tensorflow-2.0.0-beta1

tf.compat.v1.enable_eager_execution()

在 tensorflow 2.0 中，default 启用了急切执行。您不需要在您的程序中启用它。

例如

import tensorflow as tf

t = tf.constant([5.0])

现在不用session对象也可以直接查看tensor的值

print(t)
# tf.Tensor([5.], shape=(1,), dtype=float32)

也可以将张量值改为numpy数组

numpy_array = t.numpy()
print(numpy_array)
# [5.]

您还可以在 tensorflow-2 中禁用急切执行（已在 tensorflow-2.0.0-beta1 上测试。这可能不适用于未来版本。）

tf.compat.v1.disable_eager_execution()
t2 = tf.constant([5.0])
print(t2)
# Tensor("Const:0", shape=(1,), dtype=float32)

禁用急切执行后在张量上调用 numpy() 方法会引发错误

AttributeError: 'Tensor' object has no attribute 'numpy'

禁用急切执行时应该考虑的一个问题是，一旦禁用急切执行，就不能在同一个程序中启用它，因为tf.enable_eager_execution应该在程序启动时调用并在禁用后调用此方法急切执行会引发错误：

ValueError: tf.enable_eager_execution must be called at program startup.

Unable to Enable Tensorflows Eager execution