Skip to content
Snippets Groups Projects
Commit 44c3441b authored by Maxence Naud's avatar Maxence Naud
Browse files

[add] Tensor::clone binding and update Tensor display

parent 40b8dddd
No related branches found
No related tags found
3 merge requests!279v0.4.0,!253v0.4.0,!238Upd python binding
...@@ -315,6 +315,7 @@ void init_Tensor(py::module& m){ ...@@ -315,6 +315,7 @@ void init_Tensor(py::module& m){
.def(py::self - py::self) .def(py::self - py::self)
.def(py::self * py::self) .def(py::self * py::self)
.def(py::self / py::self) .def(py::self / py::self)
.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_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("set_backend", &Tensor::setBackend, py::arg("name"), py::arg("device") = 0, py::arg("copyFrom") = true)
...@@ -334,8 +335,8 @@ void init_Tensor(py::module& m){ ...@@ -334,8 +335,8 @@ void init_Tensor(py::module& m){
.def("cpy_transpose", (void (Tensor::*)(const Tensor& src, const std::vector<DimSize_t>& transpose)) &Tensor::copyTranspose, py::arg("src"), py::arg("transpose")) .def("cpy_transpose", (void (Tensor::*)(const Tensor& src, const std::vector<DimSize_t>& transpose)) &Tensor::copyTranspose, py::arg("src"), py::arg("transpose"))
.def("__str__", [](Tensor& b) { .def("__str__", [](Tensor& b) {
if (b.empty()) { if (b.empty() && b.undefined()) {
return std::string("{}"); return std::string("{}");
} else { } else {
return b.toString(); return b.toString();
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment