tf.experimental.output_all_intermediates

Выводить все промежуточные результаты от операций функционального управления потоком.

tf.experimental.output_all_intermediates(
    state
)

По умолчанию все промежуточные результаты выводятся при использовании v2 потоков управления внутри моделей Keras в режиме графа (возможно, внутри Estimators). Это необходимо для поддержки вычисления градиентов v2 потоков управления. В режиме графа Keras может заморозить прямой граф перед вычислением градиента, что не работает для v2 потоков управления, так как это требует обновления операций прямого пути для вывода необходимых промежуточных результатов. Мы обходим это, заранее вывода необходимых промежуточных результатов при построении самого прямого прохода. В идеале любые такие дополнительные тензоры должны быть удалены во время выполнения. Однако, если по какой-либо причине это не работает для вас или если у вас есть модель только для вывода, вы можете отключить это поведение, используя tf.compat.v1.experimental.output_all_intermediates(False).

Если при использовании поведения по умолчанию вы все еще видите ошибки типа "Подключение к недействительному выходу X узла-источника Y, имеющего Z выходов", попробуйте установить tf.compat.v1.experimental.output_all_intermediates(True) и отправьте отчет о проблеме на https://github.com/tensorflow/tensorflow/issues.