TensorFlow 2 version

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.

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.