diff --git a/python_binding/data/pybind_Tensor.cpp b/python_binding/data/pybind_Tensor.cpp index 067ad8c003b2f403fbf312a59f7416ad1c364a47..fa109a9af4b1146b60f0fffc80b8dfc6e4a2c256 100644 --- a/python_binding/data/pybind_Tensor.cpp +++ b/python_binding/data/pybind_Tensor.cpp @@ -71,7 +71,8 @@ void init_Tensor(py::module& m){ (m,"Tensor", py::multiple_inheritance(), py::buffer_protocol()); pyClassTensor.def(py::init<>()) - .def("set_backend", &Tensor::setBackend, py::arg("name"), py::arg("device") = 0) + .def("set_datatype", &Tensor::setDataType, py::arg("datatype"), py::arg("copyCast") = true) + .def("set_backend", &Tensor::setBackend, py::arg("name"), py::arg("device") = 0, py::arg("copyFrom") = true) .def("dims", (const std::vector<DimSize_t>& (Tensor::*)()const) &Tensor::dims) .def("dtype", &Tensor::dataType) .def("size", &Tensor::size)