torch.isinf

torch.isinf(input) → Tensor

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

Примечание

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

Параметры:

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

Возвращает:

Булев тензор, содержащий True там, где input является бесконечностью, и False в остальных случаях

Пример:

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