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