ConfusionMatrix =============== Computes the confusion matrix from predictions and labels. **Abstract Signature:** ``ConfusionMatrix(labels: Tensor, predictions: Tensor, num_classes: int)`` .. raw:: html

PyTorch

API: torch.func.confusion_matrix
Strategy: Plugin (torch_confusion_matrix_impl)

JAX (Core)

API:
Strategy: Plugin (jax_confusion_matrix_impl)

NumPy

API: sklearn.metrics.confusion_matrix
Strategy: Direct Mapping

TensorFlow

API: tf.math.confusion_matrix
Strategy: Direct Mapping