From dafd73de7fc0d51491f65b0945cc47a7eb1222fb Mon Sep 17 00:00:00 2001
From: bhalimi <benjamin.halimi@cea.fr>
Date: Fri, 17 Jan 2025 12:22:23 +0000
Subject: [PATCH] add pybind for abs and mean
---
python_binding/data/pybind_Tensor.cpp | 2 ++
python_binding/pybind_core.cpp | 2 ++
2 files changed, 4 insertions(+)
diff --git a/python_binding/data/pybind_Tensor.cpp b/python_binding/data/pybind_Tensor.cpp
index 35e60e158..487866f8a 100644
--- a/python_binding/data/pybind_Tensor.cpp
+++ b/python_binding/data/pybind_Tensor.cpp
@@ -317,6 +317,8 @@ void init_Tensor(py::module& m){
.def(py::self - py::self)
.def(py::self * py::self)
.def(py::self / py::self)
+ .def("zeros", &Tensor::zeros)
+ .def("mean", &Tensor::mean)
.def("clone", &Tensor::clone)
.def("sqrt", &Tensor::sqrt)
.def("set_datatype", &Tensor::setDataType, py::arg("datatype"), py::arg("copyCast") = true)
diff --git a/python_binding/pybind_core.cpp b/python_binding/pybind_core.cpp
index eccbebd2f..c292a8937 100644
--- a/python_binding/pybind_core.cpp
+++ b/python_binding/pybind_core.cpp
@@ -31,6 +31,7 @@ void init_Operator(py::module&);
void init_OperatorTensor(py::module&);
void init_StaticAnalysis(py::module&);
+void init_Abs(py::module&);
void init_Add(py::module&);
void init_And(py::module&);
void init_ArgMax(py::module&);
@@ -128,6 +129,7 @@ void init_Aidge(py::module& m) {
init_OperatorTensor(m);
init_StaticAnalysis(m);
+ init_Abs(m);
init_Add(m);
init_And(m);
init_ArgMax(m);
--
GitLab