From 19402c507e36efec50491a2f825d3c265ab688ae Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Thu, 7 Nov 2024 16:01:44 +0100
Subject: [PATCH] Fixed a bug in Concat with negative axis
---
src/operator/Concat.cpp | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/src/operator/Concat.cpp b/src/operator/Concat.cpp
index 55efdd51d..27b9d1cf1 100644
--- a/src/operator/Concat.cpp
+++ b/src/operator/Concat.cpp
@@ -49,7 +49,9 @@ std::shared_ptr<Aidge::Operator> Aidge::Concat_Op::clone() const {
void Aidge::Concat_OpImpl::forward() {
const Concat_Op& op = dynamic_cast<const Concat_Op&>(mOp);
- const DimSize_t axis = op.axis();
+ auto axis = op.axis();
+ const auto nbDimsInput0 = op.getInput(0)->nbDims();
+ axis = (axis < 0) ? axis + static_cast<std::int32_t>(nbDimsInput0) : axis;
assert(op.getInput(0) && "missing input in Concat operator");
for (IOIndex_t i = 1; i < mOp.nbInputs(); ++i) {
--
GitLab