jax.scipy.special.digamma#
- jax.scipy.special.digamma(x)[源代码]#
双伽马函数
scipy.special.digamma的 JAX 实现。\[\mathrm{digamma}(z) = \psi(z) = \frac{\mathrm{d}}{\mathrm{d}z}\log \Gamma(z)\]其中 \(\Gamma(z)\) 是
gamma()函数。- 参数:
x (ArrayLike) – 类数组，实数值。
- 返回:
包含双伽马函数值的数组。
- 返回类型:
笔记
JAX 版本的 digamma 接受实数值输入。