megengine.functional.isnan#

isnan(inp)[源代码]#

逐个元素的 NaN 检查。

测试输入张量 \(x\) 的每个元素 \(x_i\) ,以确定该元素是否等于 NaN

参数:

inp (Tensor) – 输入张量。 应该有一个数字数据类型。

返回类型:

Tensor

返回:

包含测试结果的张量。如果 \(x_i\)NaN ,则输出元素为 True ,否则为 False 。返回数组的数据类型应为bool。

实际案例

>>> F.isnan(Tensor(1))
Tensor(False, dtype=bool, device=xpux:0)

逐元素地判断是否是 NaN

>>> x = Tensor([1, float("nan"), 0])
>>> F.isnan(x)
Tensor([False  True False], dtype=bool, device=xpux:0)