From 55cc245923800ff299ae89ad34d28dfe380d5eb6 Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Fri, 17 Jan 2025 12:45:48 +0000
Subject: [PATCH] ADD: binding for 'Tensor::backend()'
---
python_binding/data/pybind_Tensor.cpp | 1 +
1 file changed, 1 insertion(+)
diff --git a/python_binding/data/pybind_Tensor.cpp b/python_binding/data/pybind_Tensor.cpp
index e235425d1..9e58f2a07 100644
--- a/python_binding/data/pybind_Tensor.cpp
+++ b/python_binding/data/pybind_Tensor.cpp
@@ -328,6 +328,7 @@ void init_Tensor(py::module& m){
.def("dims", (const std::vector<DimSize_t>& (Tensor::*)()const) &Tensor::dims)
.def("grad", &Tensor::grad)
.def("set_grad", &Tensor::setGrad)
+ .def("backend", &Tensor::backend)
.def("dtype", &Tensor::dataType)
.def("dformat", &Tensor::dataFormat)
.def("size", &Tensor::size)
--
GitLab