ApproxMinK

Returns min k values and their indices using approximate search.

Abstract Signature:

ApproxMinK(operand: Tensor, k: int, reduction_dimension: int = -1)

PyTorchJAX (Core)TensorFlowApple MLX

PyTorch

API: torch.topk

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.lax.approx_min_k

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.math.approx_min_k

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: —

Strategy: Plugin (mlx_approx_min_k_fallback)