megengine.functional.isinf

isinf(inp)[源代码]

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

参数

inp – 输入张量。

返回

分隔的张量。

例如:

from megengine import tensor
import megengine.functional as F

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

输出:

[False  True False]