DigammaΒΆ

Computes the logarithmic derivative of the gamma function.

Abstract Signature:

Digamma(input: Tensor)

PyTorch

API: torch.digamma
Strategy: Direct Mapping

JAX (Core)

API: jax.scipy.special.digamma
Strategy: Direct Mapping

NumPy

API: scipy.special.digamma
Strategy: Direct Mapping

Keras

API: keras.ops.digamma
Strategy: Direct Mapping

TensorFlow

API: tf.math.digamma
Strategy: Direct Mapping

Flax NNX

API: jax.scipy.special.digamma
Strategy: Direct Mapping

PaxML / Praxis

API: jax.scipy.special.digamma
Strategy: Direct Mapping