From badf5d938da491d086b8ace48a36a5a4003a97d6 Mon Sep 17 00:00:00 2001
From: Antoni Olivier <olivier.antoni@cea.fr>
Date: Thu, 10 Apr 2025 14:02:19 +0200
Subject: [PATCH] Fix update gradient tensor dimesions
---
include/aidge/data/Tensor.hpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/include/aidge/data/Tensor.hpp b/include/aidge/data/Tensor.hpp
index 5c34b0533..9b2c6d76e 100644
--- a/include/aidge/data/Tensor.hpp
+++ b/include/aidge/data/Tensor.hpp
@@ -649,7 +649,7 @@ public:
}
void setGrad(std::shared_ptr<Tensor> newGrad) {
- AIDGE_ASSERT(newGrad->dims() == mDims, "wrong dimensions");
+ AIDGE_ASSERT(!newGrad || newGrad->dims() == mDims, "wrong dimensions");
mGrad = newGrad;
}
--
GitLab