tf.math.is_finite

TensorFlow 1 version

Returns which elements of x are finite.

tf.math.is_finite(
    x, name=None
)

Example:

x = tf.constant([5.0, 4.8, 6.8, np.inf, np.nan])
tf.math.is_finite(x) ==> [True, True, True, False, False]

x A Tensor. Must be one of the following types: bfloat16, half, float32, float64.
name A name for the operation (optional).

A Tensor of type bool.

Numpy Compatibility

Equivalent to np.isfinite