add python binding for ArithmeticOperator

python_binding/operator/pybind_ArithmeticOperator.cpp

+/********************************************************************************
+ * Copyright (c) 2024 CEA-List
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License 2.0 which is available at
+ * http://www.eclipse.org/legal/epl-2.0.
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ *
+ ********************************************************************************/
+
+#include <pybind11/pybind11.h>
+#include "aidge/backend/OperatorImpl.hpp"
+#include "aidge/operator/ArithmeticOperator.hpp"
+#include "aidge/operator/Operator.hpp"
+#include <pybind11/stl.h>
+
+namespace py = pybind11;
+namespace Aidge {
+void init_ArithmeticOperator(py::module& m){
+    py::class_<ArithmeticOperator, std::shared_ptr<ArithmeticOperator>, Operator>(m, "ArithmeticOperator")
+    .def("get_output", &ArithmeticOperator::getOutput, py::arg("outputIdx"))
+    .def("get_input", &ArithmeticOperator::getInput, py::arg("inputIdx"))
+    .def("output_dims_forwarded", &ArithmeticOperator::outputDimsForwarded)
+    ;
+}
+}

python_binding/pybind_core.cpp

@@ -17,6 +17,7 @@ namespace Aidge {
 void init_Data(py::module&);
 void init_Tensor(py::module&);
 void init_OperatorImpl(py::module&);
+void init_ArithmeticOperator(py::module&);
 void init_Attributes(py::module&);
 void init_Operator(py::module&);
 void init_OperatorTensor(py::module&);
@@ -73,6 +74,7 @@ void init_Aidge(py::module& m){
     init_OpArgs(m);
     init_Connector(m);
 
+    init_ArithmeticOperator(m);
     init_OperatorImpl(m);
     init_Attributes(m);
     init_Operator(m);