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