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
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 . |
© 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/versions/r2.9/api_docs/python/tf/debugging/check_numerics