tf.debugging.check_numerics

Checks a tensor for NaN and Inf values.

tf.debugging.check_numerics(
    tensor, message, name=None
)

When run, reports an InvalidArgument error if tensor has any values that are not a number (NaN) or infinity (Inf). Otherwise, returns the input tensor.

Example usage:

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

tensor A Tensor. Must be one of the following types: bfloat16, half, float32, float64.
message A string. Prefix of the error message.
name A name for the operation (optional).

A Tensor. Has the same type as tensor.