TopKValues

Returns the k largest elements from the input (Values only).

Abstract Signature:

TopKValues(x: Tensor, k: int, axis: int = -1)

PyTorch

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

JAX (Core)

API: jax.lax.top_k
Strategy: Direct Mapping
Official Docs ↗

Keras

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

TensorFlow

API: tf.math.top_k
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

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