Fix Reshape

Context

Fix Reshape operator:

• Handle case where a dimension is 0 (see ONNX)
• Support both cases where the shape is either an input or an attribute

Modified files

• Reshape.hpp, Reshape.cpp, and pybind_Reshape.cpp;

src/operator/Reshape.cpp

@@ -54,6 +54,10 @@ bool Aidge::Reshape_Op::forwardDims(bool /*allowDataDependency*/) {
                 dimSize = 1;
                 negativeIndex = static_cast<DimIdx_t>(i);
             }
+            else if (dimSize == 0)
+            {
+                dimSize = getInput(0) -> dims()[i];
+            }
             outDims.push_back(static_cast<DimSize_t>(dimSize));
             outSize *= static_cast<DimSize_t>(dimSize);
         }