torch.isnan

torch.isnan(input) → 張量

返回一個布林型張量，其中每個元素表示輸入 input 中的對應元素是否為 NaN。當複數值的實部和/或虛部為 NaN 時，該複數值被認為是 NaN。

引數

input (張量) – 輸入張量。

返回值

一個布林型張量，其中 input 為 NaN 的位置為 True，否則為 False

示例

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