InTopK

Outputs whether the targets are in the top K predictions.

Abstract Signature:

InTopK(targets: Tensor, predictions: Tensor, k: int)

PyTorchJAX (Core)KerasTensorFlowFlax NNXPaxML / Praxis

PyTorch

API: —

Strategy: Plugin (in_top_k_plugin)

JAX (Core)

API: —

Strategy: Plugin (in_top_k_plugin)

Keras

API: keras.ops.in_top_k

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.math.in_top_k

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: —

Strategy: Plugin (in_top_k_plugin)

PaxML / Praxis

API: —

Strategy: Plugin (in_top_k_plugin)