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