diff --git a/include/aidge/operator/Transpose.hpp b/include/aidge/operator/Transpose.hpp index d6170396c940cad33e97235045361b740921ea7c..b19ec6473b1fb11ee01451e24ecb31553ff6011b 100644 --- a/include/aidge/operator/Transpose.hpp +++ b/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 { diff --git a/python_binding/operator/pybind_Transpose.cpp b/python_binding/operator/pybind_Transpose.cpp index 59895cc44a98e733badf895bcfe72bd3766c3286..2d9ce1796252974cea1535de86d5050c6af283b8 100644 --- a/python_binding/operator/pybind_Transpose.cpp +++ b/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