torch.isnan

torch.isnan(input) → Tensor

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

Параметры:

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

Возвращает:

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

Пример:

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