tf.math.is_inf

TensorFlow 2 version

Returns which elements of x are Inf.

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.isinf