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