torch.isnan

torch.isnan(input) → Tensor

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

Параметры

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

Возвращаемые значения

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

Пример:

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