From bd95076c4e2295da009544c58a973abb3edbdc22 Mon Sep 17 00:00:00 2001
From: abh2k <41710346+abh2k@users.noreply.github.com>
Date: Wed, 26 May 2021 20:10:52 +0530
Subject: [PATCH] Fixed kernal area  if ceil = true

---
 src/mlpack/methods/ann/layer/mean_pooling.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mlpack/methods/ann/layer/mean_pooling.hpp b/src/mlpack/methods/ann/layer/mean_pooling.hpp
index 8b066c6d3c..ce9252bc29 100644
--- a/src/mlpack/methods/ann/layer/mean_pooling.hpp
+++ b/src/mlpack/methods/ann/layer/mean_pooling.hpp
@@ -163,7 +163,6 @@ class MeanPooling
     arma::mat inputPre = input;
     size_t inRow = input.n_rows;
     size_t inCol = input.n_cols;
-    size_t kernalArea = kernelWidth * kernelHeight;
 
     for(int i = 1; i < inCol; i++)
       inputPre.col(i) += inputPre.col(i - 1);
@@ -186,6 +185,7 @@ class MeanPooling
         if (colEnd > input.n_cols - 1)
           colEnd = input.n_cols - 1;
 
+        size_t kernalArea = (rowEnd - rowidx + 1) * (colEnd - colidx + 1);
         val += inputPre(rowEnd, colEnd);
         if(rowidx >= 1)
         {