jax.numpy.i0

jax.numpy.i0(x)

计算第一类零阶修正贝塞尔函数。

JAX 实现的 numpy.i0()。

第一类零阶修正贝塞尔函数定义为

\[\mathrm{i0}(x) = I_0(x) = \sum_{k=0}^{\infty} \frac{(x^2/4)^k}{(k!)^2}\]

参数:

x (ArrayLike) – 标量或数组。指定贝塞尔函数的参数。不支持复数输入。

返回值:

一个数组，包含 x 的修正贝塞尔函数对应值。

返回类型:

Array

另请参阅

• jax.scipy.special.i0(): 计算零阶修正贝塞尔函数。

• jax.scipy.special.i1(): 计算一阶修正贝塞尔函数。

• jax.scipy.special.i0e(): 计算指数缩放的零阶修正贝塞尔函数。

示例

>>> x = jnp.array([-2, -1, 0, 1, 2])
>>> jnp.i0(x)
Array([2.2795851, 1.266066 , 1.0000001, 1.266066 , 2.2795851], dtype=float32)