zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add AdaptiveMaxPool1d, AdaptiveMaxPool2d, and AdaptiveMaxPool3d layers

Browse Source
This commit is contained in:
Vincent Amato
2025-08-11 23:45:52 -04:00
parent 7fde1b6a1e
commit 68c7be55bb
4 changed files with 396 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docs/src/python/nn/layers.rst
View File

@@ -9,6 +9,9 @@ Layers
:toctree: _autosummary
:template: nn-module-template.rst
AdaptiveMaxPool1d
AdaptiveMaxPool2d
AdaptiveMaxPool3d
ALiBi
AvgPool1d
AvgPool2d