torch.isinf

torch.isinf(input) → Tensor

Проверяет, является ли каждый элемент input бесконечным (положительная или отрицательная бесконечность) или нет.

Примечание

Комплексные значения являются бесконечными, когда их вещественная или мнимая часть бесконечна.

Параметры:

input (Тензор) – входной тензор.

Возвращаемое значение:

Булевый тензор, который имеет значение True там, где input бесконечен, и False в противном случае

Пример:

>>> torch.isinf(torch.tensor([1, float('inf'), 2, float('-inf'), float('nan')]))
tensor([False,  True,  False,  True,  False])