tf.contrib.eager.inf_nan_callback
Обработчик выполнения, который проверяет наличие inf и nan в выходных тензорах.
tf.contrib.eager.inf_nan_callback(
op_type, inputs, attrs, outputs, op_name, check_inf=True, check_nan=True,
action=tf.contrib.eager.ExecutionCallback.RAISE
)
Этот обработчик можно использовать с tfe.add_execute_callback для проверки на наличие недопустимых числовых значений. Например,
tfe.add_execute_callback(tfe.inf_nan_callback)
| Аргументы | |
|---|---|
op_type | Имя типа операции TFE (например, MatMul). |
inputs | list входных тензоров в операцию, в настоящее время не используется этим обработчиком. |
attrs | Атрибуты операции TFE в виде кортежа, чередующего имена и значения атрибутов. |
outputs | list выходных тензоров из операции, проверяемых этим обработчиком на наличие значений inf и nan. |
op_name | Имя операции TFE. Это имя задаётся клиентом и может быть None если не задано. |
check_inf | (bool) Нужно ли проверять значения inf в значениях выходного тензора. |
check_nan | (bool) Нужно ли проверять значения nan в значениях выходного тензора. |
action | (ExecutionCallback) Действие, которое должен выполнить обработчик при обнаружении значений inf или nan. |
| Исключения | |
|---|---|
InfOrNanError | если значения inf или nan обнаружены в любом из outputs и action имеет значение "raise". |
ValueError | если значение action является недопустимым. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/eager/inf_nan_callback