KLDivergenceLoss

Computes Kullback-Leibler divergence loss.

Abstract Signature:

KLDivergenceLoss(reduction: str = sum_over_batch_size)

PyTorch

API: torch.nn.KLDivLoss
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.losses.KLDivergence
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.keras.losses.KLDivergence
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.nn.losses.KLDivLoss
Strategy: Direct Mapping
Official Docs ↗