KthValue

Returns the k-th smallest element.

Abstract Signature:

KthValue(input: Tensor, k: int, dim: int = -1, keepdim: bool = False)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

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]'