megengine.functional.isnan

isnan(inp)[源代码]

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

参数

inp – 输入张量。

返回

分隔的张量。

例如:

from megengine import tensor
import megengine.functional as F

x = tensor([1, float("nan"), 0])
print(F.isnan(x).numpy())

输出:

[False  True False]