在 ternsorflow 2.0 中,该enable_eager_execution
方法被移至tf.compat.v1
模块。以下适用于 tensorflow-2.0.0-beta1
tf.compat.v1.enable_eager_execution()
在 tensorflow 2.0 中,eager execution默认启用。您无需在程序中启用它。
例如
import tensorflow as tf
t = tf.constant([5.0])
现在您可以直接查看张量的值,而无需使用会话对象。
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.