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