jax.numpy.isfinite

jax.numpy.isfinite(x, /)

返回一个布尔数组，指示输入的每个元素是否为有限数。

JAX 中 numpy.isfinite 的实现。

参数：

x (ArrayLike) – 输入数组或标量。

返回：

一个与 x 形状相同的布尔数组，其中 x 不是 inf，-inf 或 NaN 的地方为 True，否则为 False。

返回类型：

数组

另请参阅

• jax.numpy.isinf()：返回一个布尔数组，指示输入的每个元素是否为正无穷大或负无穷大。

• jax.numpy.isposinf()：返回一个布尔数组，指示输入的每个元素是否为正无穷大。

• jax.numpy.isneginf()：返回一个布尔数组，指示输入的每个元素是否为负无穷大。

• jax.numpy.isnan()：返回一个布尔数组，指示输入的每个元素是否为非数字 (NaN)。

示例

>>> x = jnp.array([-1, 3, jnp.inf, jnp.nan])
>>> jnp.isfinite(x)
Array([ True,  True, False, False], dtype=bool)
>>> jnp.isfinite(3-4j)
Array(True, dtype=bool, weak_type=True)