From bb2f7eafb15f2f7b8da2a193cfd15306700be306 Mon Sep 17 00:00:00 2001 From: Diego Manzanas Date: Fri, 16 Feb 2024 17:16:10 -0600 Subject: [PATCH] Fix error on MaxPooling test --- code/nnv/engine/nn/layers/MaxPooling2DLayer.m | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/code/nnv/engine/nn/layers/MaxPooling2DLayer.m b/code/nnv/engine/nn/layers/MaxPooling2DLayer.m index 3366c3ec5c..e3ec3e6ea3 100644 --- a/code/nnv/engine/nn/layers/MaxPooling2DLayer.m +++ b/code/nnv/engine/nn/layers/MaxPooling2DLayer.m @@ -371,7 +371,7 @@ function set_padding(obj, padding) % compute feature map for each cell of map do it in parallel using cpu or gpu % TODO: explore power of using GPU for this problem - maxMap = cast(zeros(1, h*w), 'like', in_image.V); % using single vector for parallel computation + maxMap = cast(zeros(1, h*w), 'like', input); % using single vector for parallel computation for l=1:h*w a = mod(l, w);