Returns which elements of x are Inf.
tf.math.is_inf(
x, name=None
)
x = tf.constant([5.0, np.inf, 6.8, np.inf]) tf.math.is_inf(x) ==> [False, True, False, True]
| Args | |
|---|---|
x | A Tensor. Must be one of the following types: bfloat16, half, float32, float64. |
name | A name for the operation (optional). |
| Returns | |
|---|---|
A Tensor of type bool. |
numpy compatibility
Equivalent to np.isinf
© 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/math/is_inf