Add asserts in Conv::forwardDims() member function

include/aidge/operator/Conv.hpp

@@ -117,6 +117,17 @@ public:
             }
             associated &= !(getInput(i)->empty());
         }
+        AIDGE_ASSERT((getInput(0)->nbDims() == (DIM+2)) &&
+                     (getInput(0)->template dims<DIM+2>()[1] == this->template getAttr<ConvAttr::InChannels>()),
+                     "Wrong input size for Conv operator.");
+        AIDGE_ASSERT((getInput(1)->nbDims() == (DIM+2)) &&
+                     (getInput(1)->template dims<DIM+2>()[1] == this->template getAttr<ConvAttr::InChannels>()) &&
+                     (getInput(1)->template dims<DIM+2>()[0] == this->template getAttr<ConvAttr::OutChannels>()),
+                     "Wrong weight size for Conv operator.");
+        if(!this->template getAttr<ConvAttr::NoBias>())
+            AIDGE_ASSERT((getInput(2)->nbDims() == (1)) &&
+                     (getInput(2)->template dims<1>()[0] == this->template getAttr<ConvAttr::OutChannels>()),
+                     "Wrong bias size for Conv operator.");
         if (associated) {
             std::array<DimSize_t, DIM + 2> outputDims{};
             const std::array<DimSize_t, DIM + 2> inputDims(getInput(0)->template dims<DIM+2>());