mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-26 02:33:21 +08:00
8 lines
121 B
ReStructuredText
8 lines
121 B
ReStructuredText
mlx.nn.MultiHeadAttention
|
|
=========================
|
|
|
|
.. currentmodule:: mlx.nn
|
|
|
|
.. autoclass:: MultiHeadAttention
|
|
|
|
|