torch.isnan

torch.isnan(input) → Tensor

返回一个新张量，其中包含布尔值，表示input中的每个元素是否为NaN。如果复数的实部或虚部是NaN，则该复数被认为是NaN。

参数

input (Tensor) – 需要输入的张量。

返回值

一个布尔张量，在input为NaN的位置上为True，其他位置为False

示例：

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