[Feat] ForwardDims Conv for NCHW and NHWC

Context

Now forward_dims works for NHWC and NCHW dataformat

Resolves issue #223 (closed)

Modified files

• feat: Conv.hpp, now Conv_Op::inChannels() checks if the Weight Producer format is [Cout,H,W,Cin] or [Cout,Cin,H,W];
• upd: Conv::forwardDims() accounts for weight data format (NHWC or NCHW)

include/aidge/operator/Conv.hpp

@@ -164,8 +164,10 @@ public:
             AIDGE_THROW_OR_ABORT(std::runtime_error, "Convolution operator has no weight Tensor associated so no specific number of input channel imposed.");
         }
         
+        // check format
         if(getInput(1)->dataFormat()==Aidge::DataFormat::NHWC) 
             return getInput(1)->template dims<DIM+2>()[DIM+1];
+        // default format is NCHW
         return getInput(1)->template dims<DIM+2>()[1];
     }
 
@@ -178,6 +180,7 @@ public:
         if (!getInput(1)) {
             AIDGE_THROW_OR_ABORT(std::runtime_error, "Convolution operator has no weight Tensor associated so no specific number of output channel imposed.");
         }
+        // first weight dimension for both NCHW (Cout,Cin,H,W) and NHWC (Cout,H,W,Cin) data format
         return getInput(1)->template dims<DIM+2>()[0];
     }