From 6c39d115bde9102e5b8069d838345efc506441bf Mon Sep 17 00:00:00 2001 From: Paul Brossier Date: Tue, 8 Jan 2019 00:05:26 +0100 Subject: [PATCH] [maxpool1d] use first element to initialize output value, add a stride --- src/ai/maxpool1d.c | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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; } -- 2.11.0