API 相关说明
返回一个新的张量,该张量表明每个元素是否为非数值类型(NaN)。
inp (Tensor) – 输入张量。
Tensor
分隔的张量。
实际案例
from megengine import tensor import megengine.functional as F x = tensor([1, float("nan"), 0]) print(F.isnan(x).numpy())
输出:
[False True False]
上一页
megengine.functional.right_shift
下一页
megengine.functional.isinf