From 5d99e71ab21dc66367d0b3e20ca4811db4c43cec Mon Sep 17 00:00:00 2001
From: bhalimi <benjamin.halimi@cea.fr>
Date: Fri, 17 Jan 2025 12:31:47 +0000
Subject: [PATCH] abs binding
---
python_binding/data/pybind_Tensor.cpp | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/python_binding/data/pybind_Tensor.cpp b/python_binding/data/pybind_Tensor.cpp
index 487866f8a..e235425d1 100644
--- a/python_binding/data/pybind_Tensor.cpp
+++ b/python_binding/data/pybind_Tensor.cpp
@@ -319,8 +319,9 @@ void init_Tensor(py::module& m){
.def(py::self / py::self)
.def("zeros", &Tensor::zeros)
.def("mean", &Tensor::mean)
+ .def("abs", &Tensor::abs)
.def("clone", &Tensor::clone)
- .def("sqrt", &Tensor::sqrt)
+ .def("sqrt", &Tensor::sqrt)
.def("set_datatype", &Tensor::setDataType, py::arg("datatype"), py::arg("copyCast") = true)
.def("set_data_format", &Tensor::setDataFormat, py::arg("data_format"), py::arg("copyTrans") = true)
.def("set_backend", &Tensor::setBackend, py::arg("name"), py::arg("device") = 0, py::arg("copyFrom") = true)
--
GitLab