megengine.functional.isinf

isinf(inp)[源代码]

逐个元素的 infinity 检查。

测试输入张量 \(x\) 的每个元素 \(x_i\) ,以确定该元素是否等于正无穷或负无穷。

参数

inp (Tensor) – 输入张量。 应该有一个数字数据类型。

返回类型

Tensor

返回

包含测试结果的张量。如果 \(x_i\) 为正无穷大或负无穷大,则输出元素为 True ,否则为 False 。返回数组的数据类型应为bool。

实际案例

>>> F.isinf(Tensor(1))
Tensor(False, dtype=bool, device=xpux:0)

逐个元素的 isinf:

>>> x = Tensor([1, float("inf"), 0])
>>> F.isinf(x)
Tensor([False  True False], dtype=bool, device=xpux:0)