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
No related merge requests found
......@@ -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