diff --git a/aidge_core/export_utils/node_export.py b/aidge_core/export_utils/node_export.py
index dff852c1660d3fb8ce35c6348d275fbce7bac8d1..df698ca8e5e415632d49a102400667ad35f173a0 100644
--- a/aidge_core/export_utils/node_export.py
+++ b/aidge_core/export_utils/node_export.py
@@ -137,6 +137,7 @@ class ExportNode(ABC):
         self.attributes["in_name"] = [None] * self.attributes["nb_in"]
         self.attributes["in_dims"] = [None] * self.attributes["nb_in"]
         self.attributes["in_dformat"] = [None] * self.attributes["nb_in"]
+        self.attributes["in_format"] = [None] * self.attributes["nb_in"]
         self.attributes["in_dtype"] = [None] * self.attributes["nb_in"]
         self.attributes["in_cdtype"] = [None] * self.attributes["nb_in"]
         self.attributes["in_chan"] = [None] * self.attributes["nb_in"]
@@ -146,6 +147,7 @@ class ExportNode(ABC):
         self.attributes["out_name"] = [None] * self.attributes["nb_out"]
         self.attributes["out_dims"] = [None] * self.attributes["nb_out"]
         self.attributes["out_dformat"] = [None] * self.attributes["nb_out"]
+        self.attributes["out_format"] = [None] * self.attributes["nb_out"]
         self.attributes["out_dtype"] = [None] * self.attributes["nb_out"]
         self.attributes["out_cdtype"] = [None] * self.attributes["nb_out"]
         self.attributes["out_chan"] = [None] * self.attributes["nb_out"]
@@ -174,6 +176,7 @@ class ExportNode(ABC):
                 self.attributes["in_name"][idx] = f"{self.attributes['name']}_input_{idx}" if parent_node is None else f"{parent_node.name()}_output_{out_id}"
                 self.attributes["in_dims"][idx] = tensor.dims()
                 self.attributes["in_dformat"][idx] = tensor.dformat()
+                self.attributes["in_format"][idx] = aidge_core.format_as(tensor.dformat())
                 self.attributes["in_dtype"][idx] = tensor.dtype()
                 self.attributes["in_cdtype"][idx] = data_conversion.aidge2c(
                     tensor.dtype())
@@ -190,6 +193,7 @@ class ExportNode(ABC):
                 self.attributes["out_name"][idx] = f"{self.attributes['name']}_output_{idx}"
                 self.attributes["out_dims"][idx] = tensor.dims()
                 self.attributes["out_dformat"][idx] = tensor.dformat()
+                self.attributes["out_format"][idx] = aidge_core.format_as(tensor.dformat())
                 self.attributes["out_dtype"][idx] = tensor.dtype()
                 self.attributes["out_cdtype"][idx] = data_conversion.aidge2c(
                     tensor.dtype())
diff --git a/python_binding/data/pybind_Data.cpp b/python_binding/data/pybind_Data.cpp
index d750c50d984dc0def5a0639ad731f41b83714823..3df61442e9403bc5f52b11416ef14adf53afb6c2 100644
--- a/python_binding/data/pybind_Data.cpp
+++ b/python_binding/data/pybind_Data.cpp
@@ -45,5 +45,8 @@ void init_Data(py::module& m){
     py::class_<Data, std::shared_ptr<Data>>(m,"Data");
 
 
+    m.def("format_as", (const char* (*)(DataType)) &format_as, py::arg("dt"));
+    m.def("format_as", (const char* (*)(DataFormat)) &format_as, py::arg("df"));
+
 }
 }