Changeset 3cbcc40 for src/ai

Timestamp:

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

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/cnn, feature/crepe

Children:

4d452fb

Parents:

6c39d11

git-author:

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

git-committer:

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

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]);
           }
         }