jax.scipy.special.hyp1f1#

jax.scipy.special.hyp1f1 = <jax._src.custom_derivatives.custom_jvp object>[源代码]#

1F1 超几何函数。

JAX 实现的 scipy.special.hyp1f1

\[\mathrm{hyp1f1}(a, b, x) = {}_1F_1(x;a, b) = \sum_{k=0}^\infty \frac{(a)_k}{(b)_kk!}x^k\]

其中 \((\cdot)_k\) 是波赫默符号 (参考 poch())。

JAX 版本仅接受正实数输入。 abx 的值,若导致 1F1 的值过大,可能会导致错误的结果;在这种情况下,请考虑启用双精度。 a = b = 0 的约定是 1,这与 scipy 的实现不同。

参数:
  • a (类数组) – 类数组,实数值

  • b (类数组) – 类数组,实数值

  • x (类数组) – 类数组,实数值

返回值:

1F1 值的数组。

返回类型:

数组