diff --git a/include/aidge/operator/ReduceMean.hpp b/include/aidge/operator/ReduceMean.hpp index 000607c60e4e3c85671e70a941bd11f3427333dd..686ffd86c3e5e40fae2186b87572a179068b5417 100644 --- a/include/aidge/operator/ReduceMean.hpp +++ b/include/aidge/operator/ReduceMean.hpp @@ -101,7 +101,7 @@ public: * @param name Name of the Operator. * @return std::shared_ptr<Node> Node containing the Operator. */ -inline std::shared_ptr<Node> ReduceMean(const std::vector<std::int32_t> &axes, +inline std::shared_ptr<Node> ReduceMean(const std::vector<std::int32_t> &axes={}, DimSize_t keep_dims=1, const std::string& name = "") { // FIXME: properly handle default w&b initialization in every cases diff --git a/include/aidge/operator/ReduceSum.hpp b/include/aidge/operator/ReduceSum.hpp index db92da1d37a9b9efc25e09a8cf8af2dc821f9626..22aa01d11612ffcec4fa3ea48f791286f105373a 100644 --- a/include/aidge/operator/ReduceSum.hpp +++ b/include/aidge/operator/ReduceSum.hpp @@ -101,7 +101,7 @@ public: * @param name Name of the Operator. * @return std::shared_ptr<Node> Node containing the Operator. */ -inline std::shared_ptr<Node> ReduceSum(const std::vector<std::int32_t> &axes, +inline std::shared_ptr<Node> ReduceSum(const std::vector<std::int32_t> &axes={}, DimSize_t keep_dims=1, const std::string& name = "") { // FIXME: properly handle default w&b initialization in every cases diff --git a/python_binding/operator/pybind_ReduceMean.cpp b/python_binding/operator/pybind_ReduceMean.cpp index 3023c077e2f3695902ca76dfa21831749f0ca82e..dae50b9af7512c5ec78f8fdf5352f527d2596bfd 100644 --- a/python_binding/operator/pybind_ReduceMean.cpp +++ b/python_binding/operator/pybind_ReduceMean.cpp @@ -40,7 +40,7 @@ void declare_ReduceMeanOp(py::module &m) { // AIDGE_ASSERT(axes.size() == DIM, "axes size [{}] does not match DIM [{}]", axes.size(), DIM); return ReduceMean(axes, keepDims, name); - }, py::arg("axes"), + }, py::arg("axes") = std::vector<std::int32_t>(), py::arg("keep_dims") = 1, py::arg("name") = ""); } diff --git a/python_binding/operator/pybind_ReduceSum.cpp b/python_binding/operator/pybind_ReduceSum.cpp index 5521909baf444b1476953e8af024f0056b1d1f4b..5a2c3eb1f5bbccce6a84e1a453047492c67b33bd 100644 --- a/python_binding/operator/pybind_ReduceSum.cpp +++ b/python_binding/operator/pybind_ReduceSum.cpp @@ -38,7 +38,7 @@ void init_ReduceSum(py::module &m) { DimSize_t keepDims, const std::string& name) { return ReduceSum(axes, keepDims, name); - }, py::arg("axes"), + }, py::arg("axes") = std::vector<std::int32_t>(), py::arg("keep_dims") = 1, py::arg("name") = ""); }