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