Pooling layers (#357)

Co-authored-by: Angelos Katharopoulos <[email protected]> Co-authored-by: Awni Hannun <[email protected]>
2024-02-12 22:08:13 -08:00
parent 40c108766b
commit e54cbb7ba6
5 changed files with 655 additions and 1 deletions
@@ -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