jax.lax.is_finite

jax.lax.is_finite(x)

逐元素 \(\mathrm{isfinite}\)。

此函数直接降低到 stablehlo.is_finite 操作。

参数:

x (ArrayLike) – 输入数组。必须具有浮点类型。

返回:

与 x 具有相同形状的布尔 dtype 数组，当 x 为 \(\pm\infty\) 或 \(\mathit{NaN}\) 时包含 False，否则包含 True。

返回类型:

数组

另请参阅

• jax.numpy.isinf()：如果数组为无穷大，则返回 True。

• jax.numpy.isnan()：如果数组为 NaN，则返回 True。