tf.debugging.assert_all_finite

Утверждает, что тензор не содержит NaN или Inf.

tf.debugging.assert_all_finite(
    x, message, name=None
)

@tf.function
def f(x):
  x = tf.debugging.assert_all_finite(x, 'Input x must be all finite')
  return x + 1

f(tf.constant([np.inf, 1, 2]))
Traceback (most recent call last):

InvalidArgumentError: ...

Аргументы
`x`	Тензор для проверки.
`message`	Сообщение для вывода при ошибке.
`name`	Имя для этой операции (необязательно).

Возвращает
Тот же тензор, что и `x`.

© 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/api_docs/python/tf/debugging/assert_all_finite