Changeset f9f03ff for src

Timestamp:

Jan 8, 2019, 12:05:26 AM (6 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/cnn_org, feature/crepe_org

Children:

7420c40

Parents:

1768cdb

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;