Skip to content
Snippets Groups Projects
Commit 24a1e8d3 authored by Maxence Naud's avatar Maxence Naud
Browse files

Fix Reshape_Op constructor binding

parent bfabda5b
No related branches found
No related tags found
No related merge requests found
......@@ -20,7 +20,7 @@ namespace Aidge {
void init_Reshape(py::module& m) {
py::class_<Reshape_Op, std::shared_ptr<Reshape_Op>, Attributes, OperatorTensor>(m, "ReshapeOp", py::multiple_inheritance())
.def(py::init<const std::vector<std::int64_t>&>(), py::arg("shape"))
.def(py::init<const std::vector<std::int64_t>&, bool>(), py::arg("shape"), py::arg("allowzero"))
.def_static("get_inputs_name", &Reshape_Op::getInputsName)
.def_static("get_outputs_name", &Reshape_Op::getOutputsName);
declare_registrable<Reshape_Op>(m, "ReshapeOp");
......
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