tf.compat.v1.experimental.output_all_intermediates

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

tf.compat.v1.experimental.output_all_intermediates(
    state
)

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

Если при поведении по умолчанию вы по-прежнему видите ошибки вида «Connecting to invalid output X of source node Y which has Z outputs», попробуйте задать tf.compat.v1.experimental.output_all_intermediates(True) и сообщите об ошибке по адресу https://github.com/tensorflow/tensorflow/issues.