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