Skip to content
Snippets Groups Projects
Commit 700c20d9 authored by Cyril Moineau's avatar Cyril Moineau Committed by Maxence Naud
Browse files

Add axis=0 as a default arg for Gather python binding.

parent 91a82930
No related branches found
No related tags found
2 merge requests!105version 0.2.0,!85Initial working python registrar.
...@@ -24,6 +24,6 @@ void init_Gather(py::module& m) { ...@@ -24,6 +24,6 @@ void init_Gather(py::module& m) {
.def("get_outputs_name", &Gather_Op::getOutputsName) .def("get_outputs_name", &Gather_Op::getOutputsName)
.def("attributes_name", &Gather_Op::staticGetAttrsName); .def("attributes_name", &Gather_Op::staticGetAttrsName);
declare_registrable<Gather_Op>(m, "GatherOp"); declare_registrable<Gather_Op>(m, "GatherOp");
m.def("Gather", &Gather, py::arg("indices"), py::arg("gathered_shape"), py::arg("axis"), py::arg("name") = ""); m.def("Gather", &Gather, py::arg("indices"), py::arg("gathered_shape"), py::arg("axis")= 0, py::arg("name") = "");
} }
} // namespace Aidge } // namespace Aidge
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