Функция is_inf
Функция is_inf применяется к тензору и проверяет
каждый его элемент на наличие бесконечных значений.
Первым параметром функция принимает входной тензор.
Вторым необязательным параметром можно передать имя операции.
Функция возвращает булев тензор той же формы, где
True соответствует бесконечному элементу,
а False - конечному.
Синтаксис
tf.math.is_inf(x, [name])
Пример
Давайте создадим тензор с бесконечными значениями
и проверим его с помощью функции is_inf:
import tensorflow as tf
t = tf.constant([1.0, float('inf'), 3.0, float('-inf'), 5.0])
res = tf.math.is_inf(t)
print(res)
Результат выполнения кода:
tf.Tensor([False True False True False], shape=(5,), dtype=bool)
Пример
Давайте применим функцию is_inf к двумерному тензору:
import tensorflow as tf
t = tf.constant([[1.0, float('inf')], [3.0, 4.0]])
res = tf.math.is_inf(t)
print(res)
Результат выполнения кода:
tf.Tensor(
[[False True]
[False False]], shape=(2, 2), dtype=bool)
Пример
Давайте проверим тензор с делением на ноль и извлечением корня из отрицательного числа:
import tensorflow as tf
t = tf.constant([1.0, 0.0, -1.0])
div = tf.divide(1.0, t)
res = tf.math.is_inf(div)
print(res)
Результат выполнения кода:
tf.Tensor([False True False], shape=(3,), dtype=bool)
