Quantile

Computes the q-th quantiles of each row of the input tensor along the dimension dim.

Abstract Signature:

Quantile(input: Tensor, q: float, dim: int = None, keepdim: bool = False, interpolation: str = 'linear')

PyTorch

API: torch.quantile
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jnp.quantile
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: np.quantile
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.ops.quantile
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tfp.stats.quantiles
Strategy: Macro 'tfp.stats.quantiles({input}, {q} * 100, axis={dim}, interpolation={interpolation})'

Apple MLX

API: mlx.core.quantile
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jnp.quantile
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: jnp.quantile
Strategy: Direct Mapping
Official Docs ↗