From f321b1d9239c57484b90dc9987e1c4d17c4b7a5f Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Mon, 1 Jul 2024 00:19:56 +0200
Subject: [PATCH] Working version
---
src/operator/Unfold.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/operator/Unfold.cpp b/src/operator/Unfold.cpp
index 5b651846b..f15bdd799 100644
--- a/src/operator/Unfold.cpp
+++ b/src/operator/Unfold.cpp
@@ -47,8 +47,8 @@ void Aidge::Unfold_OpImpl<DIM>::forward() {
for (DimSize_t n = 0; n < op.getOutput(0)->dims()[0]; ++n) {
for (DimSize_t outC = 0; outC < outChannels; ++outC) {
- const auto inOffsetH = outC % kernelDims[1];
- const auto inOffsetW = (outC / kernelDims[1]) % kernelDims[0];
+ const auto inOffsetW = outC % kernelDims[1];
+ const auto inOffsetH = (outC / kernelDims[1]) % kernelDims[0];
const auto inC = outC / kernelDims[0] / kernelDims[1];
for (DimSize_t outH = 0; outH < outHeight; ++outH) {
--
GitLab