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。 gamma 在负整数处的符号取决于从哪一侧接近极点。

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

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

参数:

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

返回:

包含伽马函数值的数组

返回类型:

数组

另请参阅

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

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

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

注意事项

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