jax.scipy.special.polygamma

jax.scipy.special.polygamma(n, x)

多伽玛函数。

scipy.special.polygamma() 的 JAX 实现。

\[\mathrm{polygamma}(n, x) = \psi^{(n)}(x) = \frac{\mathrm{d}^n}{\mathrm{d}x^n}\log \Gamma(x)\]

其中 \(\Gamma\) 是 gamma() 函数。

参数:

• n (ArrayLike) – 类数组，整数值。导数的阶数。

• x (ArrayLike) – 类数组，实数值。用于评估函数的值。

返回:

数组

返回类型:

Array

另请参阅

• jax.scipy.special.gamma()

• jax.scipy.special.digamma()