diff --git a/python_binding/operator/pybind_Operator.cpp b/python_binding/operator/pybind_Operator.cpp
index 559c4b0d6a4c8dd2f7cb7429e663e95a058d7f20..f9482eda2f93b5492cfcc89175da69d140f23df8 100644
--- a/python_binding/operator/pybind_Operator.cpp
+++ b/python_binding/operator/pybind_Operator.cpp
@@ -27,7 +27,6 @@ void init_Operator(py::module& m){
     .def("nb_data", &Operator::nbData)
     .def("nb_param", &Operator::nbParam)
     .def("nb_outputs", &Operator::nbOutputs)
-    .def("output_dims_forwarded", &Operator::outputDimsForwarded)
     .def("associate_input", &Operator::associateInput, py::arg("inputIdx"), py::arg("data"))
     .def("set_datatype", &Operator::setDataType, py::arg("dataType"))
     .def("set_backend", &Operator::setBackend, py::arg("name"))
diff --git a/python_binding/operator/pybind_OperatorTensor.cpp b/python_binding/operator/pybind_OperatorTensor.cpp
index 605ee27f491b02e5b4eeeb77287f5332bfd20bb0..f5c95b2c6d1ba0ae7b23e9748896b26f973333a7 100644
--- a/python_binding/operator/pybind_OperatorTensor.cpp
+++ b/python_binding/operator/pybind_OperatorTensor.cpp
@@ -21,6 +21,7 @@ void init_OperatorTensor(py::module& m){
     py::class_<OperatorTensor, std::shared_ptr<OperatorTensor>, Operator>(m, "OperatorTensor")
     .def("get_output", &OperatorTensor::getOutput, py::arg("outputIdx"))
     .def("get_input", &OperatorTensor::getInput, py::arg("inputIdx"))
+    .def("output_dims_forwarded", &Operator::outputDimsForwarded)
     ;
 }
 }