KthValue ======== Returns the k-th smallest element. **Abstract Signature:** ``KthValue(input: Tensor, k: int, dim: int = -1, keepdim: bool = False)`` .. raw:: html

PyTorch

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

JAX (Core)

API: —
Strategy: Macro 'jnp.sort({input}, axis={dim})[..., {k}-1]'

NumPy

API: —
Strategy: Macro 'np.sort({input}, axis={dim})[..., {k}-1]'

Keras

API: —
Strategy: Macro 'keras.ops.sort({input}, axis={dim})[..., {k}-1]'

TensorFlow

API: —
Strategy: Macro 'tf.sort({input}, axis={dim})[..., {k}-1]'

Apple MLX

API: —
Strategy: Macro 'mx.sort({input}, axis={dim})[..., {k}-1]'

Flax NNX

API: —
Strategy: Macro 'jnp.sort({input}, axis={dim})[..., {k}-1]'

PaxML / Praxis

API: —
Strategy: Macro 'jnp.sort({input}, axis={dim})[..., {k}-1]'