jax.scipy.special.polygamma

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

多伽玛函数。

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

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

其中 \(\psi\) 是 digamma() 函数，\(\Gamma\) 是 gamma() 函数。

参数:

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

• x (ArrayLike) – 数组类，实数值。计算函数的点。

返回:

array

返回类型:

Array

另请参阅

• jax.scipy.special.gamma()

• jax.scipy.special.digamma()