scaled_dot_product_attention

Auto-generated from mlx_code_defs

PyTorch

API: torch.nn.functional.scaled_dot_product_attention
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.fast.scaled_dot_product_attention
Strategy: Direct Mapping
Official Docs ↗