PReLU layer for multidimensional input

2026-07-29 15:23:05 +04:00 · 2017-10-23 14:30:40 +03:00
parent cca99bf824
commit a36ebaecdc
4 changed files with 10 additions and 11 deletions
@@ -242,9 +242,8 @@ public:
                    }
                }

-                // TODO: check whether this is correct in the case of ChannelsPReLU.
                if(activ)
-                    activ->forwardSlice(dptr, dptr, nw, 0, 0, 1);
+                    activ->forwardSlice(dptr, dptr, 1, 1, delta, delta + nw);

                ofs += nw;
            }