megengine.functional.isnan

isnan(inp)[源代码]

返回一个新的张量,该张量表明每个元素是否为非数值类型(NaN)。

参数

inp (Tensor) – 输入张量。

返回类型

Tensor

返回

分隔的张量。

实际案例

>>> x = Tensor([1, float("nan"), 0])
>>> F.isnan(x).numpy()
array([False,  True, False])