megengine.functional.isinf¶

返回一个张量，它表示每个元素是否是无穷大值(Inf)。

实际案例

from megengine import tensor
import megengine.functional as F

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

输出：

[False  True False]

megengine.functional.isnan

megengine.functional.logical_and