jax.scipy.special.digamma

jax.scipy.special.digamma(x)

二阶对数-伽马函数

JAX 对 scipy.special.digamma 的实现。

\[\mathrm{digamma}(z) = \psi(z) = \frac{\mathrm{d}}{\mathrm{d}z}\log \Gamma(z)\]

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

参数:

x (ArrayLike) – 类数组，实数值。

返回:

包含二阶对数-伽马函数值的数组。

返回类型:

Array

注意事项

JAX 版本的 digamma 接受实数值输入。

另请参阅

• jax.scipy.special.gamma()

• jax.scipy.special.polygamma()