From: Paul Brossier <piem@piem.org>
Date: Mon, 7 Jan 2019 23:05:26 +0000 (+0100)
Subject: [maxpool1d] use first element to initialize output value, add a stride
X-Git-Url: https://git.aubio.org/?a=commitdiff_plain;h=6c39d115bde9102e5b8069d838345efc506441bf;p=aubio.git

[maxpool1d] use first element to initialize output value, add a stride
---

diff --git a/src/ai/maxpool1d.c b/src/ai/maxpool1d.c
index f171faca..58dbf68a 100644
--- a/src/ai/maxpool1d.c
+++ b/src/ai/maxpool1d.c
@@ -90,9 +90,10 @@ void aubio_maxpool1d_do(aubio_maxpool1d_t *c, aubio_tensor_t *input_tensor,
 
   for (j = 0; j < output_tensor->shape[1]; j++) {
     for (i = 0; i < output_tensor->shape[0]; i++) {
-      smpl_t m = -FLT_MAX;
+      uint_t stride_i = i * c->pool_size;
+      smpl_t m = input_tensor->data[stride_i][j];
       for (a = 0; a < c->pool_size; a++) {
-        m = MAX(m, input_tensor->data[i * c->pool_size + a][j]);
+        m = MAX(m, input_tensor->data[stride_i + a][j]);
       }
       output_tensor->data[i][j] = m;
     }