cross_entropyΒΆ

Auto-generated from mlx_code_defs

PyTorch

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

Apple MLX

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