Checks a tensor for NaN, -Inf and +Inf values.
tf.raw_ops.CheckNumericsV2( 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, passes tensor
as-is. Unlike CheckNumerics (V1), CheckNumericsV2 distinguishes -Inf and +Inf in the errors it throws.
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 . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/CheckNumericsV2