Add padding to avg pooling

2025-11-05 17:34:21 +00:00 · 2024-05-26 18:54:12 +02:00
parent d1fc45d9e0
commit 4a67b708f0
6 changed files with 60 additions and 11 deletions
--- a/include/layers/avg_pooling.cuh
+++ b/include/layers/avg_pooling.cuh
@@ -13,6 +13,7 @@ class AvgPooling2d : public SequentialLayer, public TwoDLayer {
        int            nChannels,
        dim2d           poolingSize,
        dim2d           stride,
+        dim2d           padding,
        ActivationType activationType
    );
    ~AvgPooling2d();
@@ -40,6 +41,7 @@ class AvgPooling2d : public SequentialLayer, public TwoDLayer {
    int  nChannels;
    dim2d poolingSize;
    dim2d stride;
+    dim2d padding;

    dim2d outputSize;