Skip to content
Snippets Groups Projects

Chore_26_compare-layer-import-export-onnx

Merged Maxence Naud requested to merge chore_26-compare-layer-import-export-onnxruntime into dev
2 files
+ 11
3
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -51,6 +51,7 @@ void addCtor(py::class_<Tensor,
return newTensor;
}), py::arg("array"), py::arg("backend")="cpu")
.def(py::init<T>(), py::arg("val"))
.def("__setitem__", (void (Tensor::*)(std::size_t, T)) &Tensor::set)
.def("__setitem__", (void (Tensor::*)(std::vector<std::size_t>, T)) &Tensor::set)
;
@@ -73,6 +74,7 @@ void init_Tensor(py::module& m){
(m,"Tensor", py::multiple_inheritance(), py::buffer_protocol());
pyClassTensor.def(py::init<>())
.def(py::init<const std::vector<std::size_t>&>(), py::arg("dims"))
.def(py::self + py::self)
.def(py::self - py::self)
.def(py::self * py::self)
@@ -86,7 +88,7 @@ void init_Tensor(py::module& m){
.def("dtype", &Tensor::dataType)
.def("size", &Tensor::size)
.def("capacity", &Tensor::capacity)
.def("resize", (void (Tensor::*)(const std::vector<DimSize_t>&, std::vector<DimSize_t>)) &Tensor::resize)
.def("resize", (void (Tensor::*)(const std::vector<DimSize_t>&, std::vector<DimSize_t>)) &Tensor::resize, py::arg("dims"), py::arg("strides") = std::vector<DimSize_t>())
.def("has_impl", &Tensor::hasImpl)
.def("get_coord", &Tensor::getCoord)
.def("get_idx", &Tensor::getIdx)
Loading