categorical_crossentropyΒΆ

Computes the categorical crossentropy loss.

Abstract Signature:

categorical_crossentropy(y_true, y_pred, from_logits: bool = False)

PyTorch

API: torch.nn.functional.cross_entropy
Strategy: Direct Mapping

JAX (Core)

API: optax.softmax_cross_entropy
Strategy: Direct Mapping

Keras

API: keras.losses.categorical_crossentropy
Strategy: Direct Mapping

TensorFlow

API: tf.keras.losses.categorical_crossentropy
Strategy: Direct Mapping

Apple MLX

API: mlx.nn.losses.cross_entropy
Strategy: Direct Mapping