mirror of
https://github.com/ml-explore/mlx.git
synced 2025-06-24 17:31:16 +08:00
24 lines
344 B
ReStructuredText
24 lines
344 B
ReStructuredText
|
.. _nn_functions:
|
||
|
|
|
||
|
|
.. currentmodule:: mlx.nn
|
||
|
|
|
||
|
|
Functions
|
||
|
|
---------
|
||
|
|
|
||
|
|
Layers without parameters (e.g. activation functions) are also provided as
|
||
|
|
simple functions.
|
||
|
|
|
||
|
|
.. autosummary::
|
||
|
|
:toctree: _autosummary_functions
|
||
|
|
:template: nn-module-template.rst
|
||
|
|
|
||
|
|
gelu
|
||
|
|
gelu_approx
|
||
|
|
gelu_fast_approx
|
||
|
|
relu
|
||
|
|
prelu
|
||
|
|
silu
|
||
|
|
step
|
||
|
|
selu
|
||
|
|
mish
|