Pooling: fix bug with non-square pooling size

Maratyszcza · Mar 17, 2017 · 1a75285 · 1a75285
1 parent 4d34ea4
commit 1a75285
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pooling-output.c b/src/pooling-output.c
@@ -48,7 +48,7 @@ static void compute_max_pooling_forward__generic(
 				const size_t s = y * stride_height + i - padding_top;
 				if (s < input_height) {
 					for (size_t j = 0; j < pooling_width; j++) {
-						const size_t t = x * stride_height + j - padding_left;
+						const size_t t = x * stride_width + j - padding_left;
 						if (t < input_width) {
 							v = maxf(input[s][t], v);
 						}