LBeta

Computes ln(|Beta(x)|), reducing along the last dimension.

Abstract Signature:

LBeta(x: Tensor)

PyTorchJAX (Core)KerasTensorFlowFlax NNXPaxML / Praxis

PyTorch

API: torch.lbeta

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro 'jax.scipy.special.gammaln({x}).sum(-1) - jax.scipy.special.gammaln({x}.sum(-1))'

Keras

API: —

Strategy: Macro 'keras.ops.sum(keras.ops.log_gamma({x}), axis=-1) - keras.ops.log_gamma(keras.ops.sum({x}, axis=-1))'

TensorFlow

API: tf.math.lbeta

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: —

Strategy: Macro 'jax.scipy.special.gammaln({x}).sum(-1) - jax.scipy.special.gammaln({x}.sum(-1))'

PaxML / Praxis

API: —

Strategy: Macro 'jax.scipy.special.gammaln({x}).sum(-1) - jax.scipy.special.gammaln({x}.sum(-1))'