jax.scipy.special.polygamma

内容

jax.scipy.special.polygamma#

jax.scipy.special.polygamma(n, x)[source]#

多伽马函数。

JAX 实现 scipy.special.polygamma()

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

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

参数:
  • n (ArrayLike) – 类数组,整数值。导数的阶数。

  • x (ArrayLike) – 类数组,实数值。要评估函数的值。

返回值:

数组

返回类型:

数组