tf.check_numerics

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

Defined in tensorflow/python/ops/gen_array_ops.py.

See the guide: Control Flow > Debugging Operations

Checks a tensor for NaN and Inf values.

When run, reports an InvalidArgument error if tensor has any values that are not a number (NaN) or infinity (Inf). Otherwise, passes tensor as-is.

Args:

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

Returns:

A Tensor. Has the same type as tensor.