add more Registrar dims for Transpose

include/aidge/operator/Transpose.hpp

@@ -103,19 +103,19 @@ class Transpose_Op : public Operator,
 
 
     inline std::shared_ptr<Tensor> getInput(const IOIndex_t inputIdx) const override final {
-        assert((inputIdx == 0)  && "Transpose Operators supports only 1 input");
+        assert((inputIdx == 0)  && "Transpose operator supports only 1 input");
         (void) inputIdx; // avoid unused warning
         return mInput;
     }
     inline std::shared_ptr<Tensor> getOutput(const IOIndex_t outputIdx) const override final {
-        assert((outputIdx == 0) && "Transpose Operator has only 1 output");
+        assert((outputIdx == 0) && "Transpose operator has only 1 output");
         (void) outputIdx; // avoid unused warning
         return mOutput;
     }
 
 
     std::shared_ptr<Data> getRawInput(const IOIndex_t inputIdx) const override final {
-        assert((inputIdx == 0) && "Transpose Operator supports only 1 input");
+        assert((inputIdx == 0) && "Transpose operator supports only 1 input");
         return std::static_pointer_cast<Data>(mInput);
     }
     std::shared_ptr<Data> getRawOutput(const IOIndex_t outputIdx) const override final {

python_binding/operator/pybind_Transpose.cpp

@@ -26,7 +26,7 @@ namespace py = pybind11;
 namespace Aidge {
 
 template <DimIdx_t DIM> 
-void init_Transpose(py::module &m) {
+void declare_Transpose(py::module &m) {
   py::class_<Transpose_Op<DIM>, std::shared_ptr<Transpose_Op<DIM>>, Operator, Attributes>(
     m, ("TransposeOp" + std::to_string(DIM) + "D").c_str(), py::multiple_inheritance())
   .def("get_inputs_name", &Transpose_Op<DIM>::getInputsName)
@@ -41,4 +41,12 @@ void init_Transpose(py::module &m) {
 
 }
 
+void init_Transpose(py::module &m) {
+  declare_Transpose<2>(m);
+  declare_Transpose<3>(m);
+  declare_Transpose<4>(m);
+  declare_Transpose<5>(m);
+  declare_Transpose<6>(m);
+
+}
 } // namespace Aidge