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