tf.compat.v1.enable_control_flow_v2

Используйте поток управления v2.

tf.compat.v1.enable_control_flow_v2()

Поток управления v2 (cfv2) — это улучшенная версия потока управления в TensorFlow с поддержкой производных высших порядков. Включение cfv2 изменит представление потока управления в графах и функциях: например, tf.while_loop и tf.cond будут создавать функциональные операции While и If вместо низкоуровневых операций Switch, Merge и т. д. Примечание: импорт и выполнение графов, экспортированных с использованием старого потока управления, по-прежнему поддерживаются.

Вызов tf.enable_control_flow_v2() позволяет включить эту функцию TensorFlow 2.0.

Примечание: поток управления v2 всегда включен внутри tf.function. Вызывать эту функцию не требуется.