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])