tensorflow - 无法启用 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)

也可以将tensor值改为numpy数组

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

您还可以在 tensorflow-2 中禁用 eager execution(已在 tensorflow-2.0.0-beta1 上测试。这可能不适用于 future 版本。)

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.