jax.numpy.i0

jax.numpy.i0(x)

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

numpy.i0() 的 JAX 实现。

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

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

参数:

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

返回:

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

返回类型:

数组

另请参阅

• 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)