jax.numpy.isinf

jax.numpy.isinf(x, /)

逐元素测试正无穷或负无穷。

LAX 后端实现 numpy.isinf()。

原始文档字符串如下。

返回与 x 形状相同的布尔数组，在 x == +/-inf 处为 True，否则为 False。

参数：

x (array_like) – 输入值

返回值：

y – 在 x 为正无穷或负无穷时为 True，否则为 false。如果 x 是标量，则为标量。

返回类型：

bool（标量）或布尔 ndarray