megengine.functional.isnan¶

逐个元素的 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)

megengine.functional.right_shift

megengine.functional.isinf