tf.raw_ops.CheckNumerics

Проверяет тензор на наличие значений NaN и Inf.

tf.raw_ops.CheckNumerics(
    tensor, message, name=None
)

При выполнении, сообщает об ошибке InvalidArgument, если tensor содержит значения, которые не являются числами (NaN) или бесконечностью (Inf). В противном случае возвращает входной тензор.

Пример использования:

a = tf.Variable(1.0)
tf.debugging.check_numerics(a, message='')

b = tf.Variable(np.nan)
try:
  tf.debugging.check_numerics(b, message='Checking b')
except Exception as e:
  assert "Checking b : Tensor had NaN values" in e.message

c = tf.Variable(np.inf)
try:
  tf.debugging.check_numerics(c, message='Checking c')
except Exception as e:
  assert "Checking c : Tensor had Inf values" in e.message