Python tf.compat.v1.enable_eager_execution用法及代码示例

在此程序的生命周期内启用即刻执行。

用法

tf.compat.v1.enable_eager_execution(
    config=None, device_policy=None, execution_mode=None
)

参数

config (可选。)tf.compat.v1.ConfigProto 用于配置执行操作的环境。请注意，tf.compat.v1.ConfigProto 还用于配置图形执行(通过 tf.compat.v1.Session )，当启用即刻执行时，tf.compat.v1.ConfigProto 中的许多选项都没有实现(或不相关)。
device_policy
(可选。)控制需要特定设备(例如 GPU 0)上的输入的操作如何处理不同设备(例如 GPU 1 或 CPU)上的输入的策略。当设置为无时，将自动选择适当的值。选择的值可能会在 TensorFlow 版本之间发生变化。有效值：
- tf.contrib.eager.DEVICE_PLACEMENT_EXPLICIT：如果放置不正确，则会引发错误。
- tf.contrib.eager.DEVICE_PLACEMENT_WARN：复制不在正确设备上但记录警告的张量。
- tf.contrib.eager.DEVICE_PLACEMENT_SILENT: 默默地复制张量。请注意，这可能会隐藏性能问题，因为当在设备之间复制的张量上的操作被阻止时不会提供通知。
- tf.contrib.eager.DEVICE_PLACEMENT_SILENT_FOR_INT32：默默地复制 int32 张量，在其他张量上引发错误。
execution_mode
(可选。)控制调度操作如何实际执行的策略。当设置为无时，将自动选择适当的值。选择的值可能会在 TensorFlow 版本之间发生变化。有效值：
- tf.contrib.eager.SYNC：同步执行每个操作。
- tf.contrib.eager.ASYNC：异步执行每个操作。这些操作可能会返回 "non-ready" 句柄。

抛出

ValueError 如果在创建/执行 TensorFlow 图后启用了即刻执行，或者如果提供的选项与之前对该函数的调用发生冲突。

迁移到 TF2

警告：这个 API 是为 TensorFlow v1 设计的。继续阅读有关如何从该 API 迁移到本机 TensorFlow v2 等效项的详细信息。见TensorFlow v1 到 TensorFlow v2 迁移指南有关如何迁移其余代码的说明。

如果您使用的是 TF2，则不需要此函数。默认情况下启用即刻执行。

Eager execution 为 TensorFlow 提供了一个命令式接口。启用即刻执行后，TensorFlow 函数会立即执行操作(而不是添加到稍后在 tf.compat.v1.Session 中执行的图形)并返回具体值(而不是对计算图中节点的符号引用)。

例如：

tf.compat.v1.enable_eager_execution()

# After eager execution is enabled, operations are executed as they are
# defined and Tensor objects hold concrete values, which can be accessed as
# numpy.ndarray`s through the numpy() method.
assert tf.multiply(6, 7).numpy() == 42

使用 TensorFlow API 创建或执行图表后，无法启用即刻执行。通常建议在程序启动时调用此函数，而不是在库中调用(因为大多数库在有或没有即刻执行的情况下都应该可用)。

相关用法

注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.compat.v1.enable_eager_execution。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。