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 不支持复数值输入。