From 20f3c105dfa0c0c1038da47ba1ab14ec4b65615d Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Tue, 13 Feb 2024 15:15:00 +0000
Subject: [PATCH] [Fix] input unsqueeze during output dimensions computation
process in MatMul_Op
---
src/operator/MatMul.cpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/operator/MatMul.cpp b/src/operator/MatMul.cpp
index 4c7195af2..f48c7ca81 100644
--- a/src/operator/MatMul.cpp
+++ b/src/operator/MatMul.cpp
@@ -47,10 +47,10 @@ void Aidge::MatMul_Op::computeOutputDims() {
if (dims0.size() > dims1.size()) {
- dims1.insert(dims1.cbegin(), dims0.begin(), dims0.end() - dims1.size());
+ dims1.insert(dims1.cbegin(), dims0.size() - dims1.size(), std::size_t(1));
}
else if (dims1.size() > dims0.size()) {
- dims0.insert(dims0.cbegin(), dims1.begin(), dims1.end() - dims0.size());
+ dims0.insert(dims0.cbegin(), dims1.size() - dims0.size(), std::size_t(1));
}
AIDGE_ASSERT(dims0[dims_size-1] == dims1[dims_size-2], "Incompatible matrices sizes.");
--
GitLab