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

diff --git a/src/ai/maxpool1d.c b/src/ai/maxpool1d.c
index f171fac..58dbf68 100644 (file)
--- 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;
     }