Skip to content
Snippets Groups Projects
Commit 261e60e0 authored by Cyril Moineau's avatar Cyril Moineau
Browse files

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

parent fe7ad6ba
No related branches found
No related tags found
2 merge requests!105version 0.2.0,!85Initial working python registrar.
Pipeline #40509 passed
......@@ -24,6 +24,6 @@ void init_Gather(py::module& m) {
.def("get_outputs_name", &Gather_Op::getOutputsName)
.def("attributes_name", &Gather_Op::staticGetAttrsName);
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
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