megengine.functional.isinf

isinf(inp)[源代码]

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

参数

inp (Tensor) – 输入张量。

返回类型

Tensor

返回

分隔的张量。

实际案例

>>> x = Tensor([1, float("inf"), 0])
>>> F.isinf(x).numpy()
array([False,  True, False])