jax.lax.is_finite

jax.lax.is_finite(x)

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

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

参数:

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

返回:

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

返回类型:

数组

另请参阅

• jax.numpy.isinf()：当数组为无穷大时返回 True。

• jax.numpy.isnan()：当数组为 NaN 时返回 True。