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 的修正贝塞尔函数相应值的数组。

返回类型:

数组

另请参阅

示例

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