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