jax.scipy.special.gamma

jax.scipy.special.gamma(x)

伽马函数。

scipy.special.gamma 的 JAX 实现。

伽马函数定义为 \(\Re(z)>0\) 的

\[\mathrm{gamma}(z) = \Gamma(z) = \int_0^\infty t^{z-1}e^{-t}\mathrm{d}t\]

并通过解析延拓扩展到任意复数值 z。对于正整数 n，伽马函数通过以下恒等式与 factorial() 函数相关

\[\Gamma(n) = (n - 1)!\]

• 如果 \(z = -\infty\)，则返回 NaN。

• 如果 \(x = \pm 0\)，则返回 \(\pm \infty\)。

• 如果 \(x\) 是负整数，则返回 NaN。负整数处伽马函数的符号取决于从哪个方向接近极点。

• 如果 \(x = \infty\)，则返回 \(\infty\)。

• 如果 \(x\) 是 NaN，则返回 NaN。

参数:

x (ArrayLike) – 类似数组，实数值。

返回:

包含伽马函数值的数组

返回类型:

数组

另请参阅

• jax.scipy.special.factorial()：阶乘函数。

• jax.scipy.special.gammaln()：伽玛函数的自然对数

• jax.scipy.special.gammasgn()：伽玛函数的符号

注释

与 SciPy 版本不同，JAX 的 gamma 不支持复数值输入。