Python PyTorch isnan用法及代碼示例

本文簡要介紹python語言中 torch.isnan 的用法。

用法:

torch.isnan(input) → Tensor

參數：

input(Tensor) -輸入張量。

返回：

布爾張量，其中 input 為 NaN 為 True，其他地方為 False

返回一個新張量，其中布爾元素表示 input 的每個元素是否為 NaN。當複數值的實部和/或虛部為 NaN 時，將被視為NaN。

例子：

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