Pooling layers (#357)

Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com> Co-authored-by: Awni Hannun <awni@apple.com>
2025-10-31 16:21:27 +08:00 · 2024-02-13 07:08:13 +01:00
parent 40c108766b
commit e54cbb7ba6
5 changed files with 655 additions and 1 deletions
--- a/docs/src/python/nn/layers.rst
+++ b/docs/src/python/nn/layers.rst
@@ -10,6 +10,8 @@ Layers
   :template: nn-module-template.rst

   ALiBi
+   AvgPool1d
+   AvgPool2d
   BatchNorm
   Conv1d
   Conv2d
@@ -22,6 +24,8 @@ Layers
   InstanceNorm
   LayerNorm
   Linear
+   MaxPool1d
+   MaxPool2d
   Mish
   MultiHeadAttention
   PReLU