jax.numpy.isfinite

jax.numpy.isfinite(x, /)

逐元素测试是否为有限数（非无穷大和非非数字）。

LAX 后端实现 numpy.isfinite()。

原始文档字符串如下。

结果以布尔数组的形式返回。

参数：

x (array_like) – 输入值。

返回值：

y – 在 x 不是正无穷大、负无穷大或 NaN 时为 True；否则为 False。如果 x 是标量，则为标量。

返回类型：

ndarray, bool