Skip to content
Snippets Groups Projects
Commit 5d99e71a authored by Benjamin Halimi's avatar Benjamin Halimi
Browse files

abs binding

parent dafd73de
No related branches found
No related tags found
2 merge requests!318[Upd] release verision 0.5.0,!303Add pybind for Abs and Mean
......@@ -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)
......
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