Changeset 6c39d11

Timestamp:

Dec 29, 2021, 5:51:48 PM (3 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/cnn, feature/crepe

Children:

3cbcc40

Parents:

b9616b0

git-author:

Paul Brossier <piem@piem.org> (01/08/19 00:05:26)

git-committer:

Paul Brossier <piem@piem.org> (12/29/21 17:51:48)

Message:

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

File:

• src/ai/maxpool1d.c (modified) (1 diff)

src/ai/maxpool1d.c

@@ -91 +91 @@
   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;