torch.isfinite

torch.isfinite(input) → Tensor

Возвращает новый тензор с логическими элементами, представляющими, является ли каждый элемент finite или нет.

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

Параметры

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

Возвращает

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

Пример:

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