diff --git a/python_binding/data/pybind_DataFormat.cpp b/python_binding/data/pybind_DataFormat.cpp index 5308fb3023e15a74f0dc5f674917d7ae65cbb52f..d412511532a563182b40776174892d298a742b45 100644 --- a/python_binding/data/pybind_DataFormat.cpp +++ b/python_binding/data/pybind_DataFormat.cpp @@ -67,6 +67,7 @@ void init_DataFormat(py::module& m) { bindEnum<DataFormat>(m, "dformat"); m.def("format_as", (const char* (*)(DataFormat)) &format_as, py::arg("df")); m.def("get_permutation_mapping", &getPermutationMapping, py::arg("src"), py::arg("dst")); + m.def("get_transpose_output_data_format", &getTransposeOutputDataFormat, py::arg("inputDataFormat"), py::arg("outputDimsOrder")); } } // namespace Aidge