Checks a tensor for NaN and Inf values.
tf.raw_ops.CheckNumerics( 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 Args | |
|---|---|
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). |
Returns | |
|---|---|
A Tensor. Has the same type as tensor. |