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).
Если при стандартном поведении вы по-прежнему видите ошибки вида «Подключение к недействительному выводу X узла-источника Y, имеющего Z выходов», попробуйте установить tf.compat.v1.experimental.output_all_intermediates(True) и, пожалуйста, создайте вопрос на https://github.com/tensorflow/tensorflow/issues.
| Аргументы | |
|---|---|
state | True, False или None. None восстанавливает поведение по умолчанию. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/compat/v1/experimental/output_all_intermediates