Add padding to max pooling

2025-11-06 01:34:22 +00:00 · 2024-05-26 19:03:10 +02:00
parent 4a67b708f0
commit 94a16b4352
7 changed files with 78 additions and 19 deletions
--- a/include/layers/max_pooling.cuh
+++ b/include/layers/max_pooling.cuh
@@ -13,6 +13,7 @@ class MaxPooling2d : public SequentialLayer, public TwoDLayer {
        int            nChannels,
        dim2d          poolingSize,
        dim2d          stride,
+        dim2d          padding,
        ActivationType activationType
    );
    ~MaxPooling2d();
@@ -40,6 +41,7 @@ class MaxPooling2d : public SequentialLayer, public TwoDLayer {
    int   nChannels;
    dim2d poolingSize;
    dim2d stride;
+    dim2d padding;

    dim2d outputSize;