diff --git a/python_binding/data/pybind_Tensor.cpp b/python_binding/data/pybind_Tensor.cpp index e235425d134dd96c94a7b577610b3905d682dfae..9e58f2a07ca73d65f15290982ae833ac2dc3b4c9 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)