Softmax2d

Auto-generated from mlx_code_defs

PyTorch

API: torch.nn.modules.activation.Softmax2d
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.softmax
Strategy: Direct Mapping
Official Docs ↗