Changeset 7420c40

Timestamp:

Jan 8, 2019, 12:06:02 AM (5 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/cnn_org, feature/crepe_org

Children:

8691154

Parents:

f9f03ff

Message:

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

File:

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

src/ai/maxpool2d.c

@@ -102 +102 @@
       for (k = 0; k < output_tensor->shape[2]; k++)
       {
-        smpl_t m = -10000.;
+        uint_t stride_i = i * c->pool_size[0];
+        uint_t stride_j = j * c->pool_size[1];
+        smpl_t m = input_tensor->data[stride_i][stride_j
+          * input_tensor->shape[2] + k];
         for (a = 0; a < c->pool_size[0]; a++) {
           for (b = 0; b < c->pool_size[1]; b++) {
-            uint_t idx = (j * c->pool_size[1] + b) * input_tensor->shape[2] + k;
-            m = MAX(m, input_tensor->data[i * c->pool_size[0] + a][idx]);
+            uint_t idx = (stride_j + b) * input_tensor->shape[2] + k;
+            m = MAX(m, input_tensor->data[stride_i + a][idx]);
           }
         }