megengine.functional.isinf¶
- isinf(inp)[源代码]¶
逐个元素的
infinity检查。测试输入张量 \(x\) 的每个元素 \(x_i\) ，以确定该元素是否等于正无穷或负无穷。
- 参数
inp (
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)