torch.isfinite

torch.isfinite(input) → Tensor

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

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

Параметры:

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

Возвращает:

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

Пример:

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