diff --git a/include/aidge/operator/ReduceMean.hpp b/include/aidge/operator/ReduceMean.hpp index 5d5895a8fb279f1efa5c6321614199f44402b83a..16fc08af0aab86cfd09ed66486dbc4dfe834c702 100644 --- a/include/aidge/operator/ReduceMean.hpp +++ b/include/aidge/operator/ReduceMean.hpp @@ -52,12 +52,12 @@ public: /** * @brief constructor for ReduceMean op * @param[in] axes around which perform the operation - * @param[in] keep_dims if true we set a dimension of 1 in the place of the reduced axes and + * @param[in] keep_dims if true we set a dimension of 1 in the place of the reduced axes and * if false we remove the dimension completely * @param[in] noop_with_empty_axes used when no axes are provided, if set to true, the operator does nothing * and if false, we reduce on all axes */ - ReduceMean_Op(const std::vector<std::int32_t>& axes, bool keep_dims, bool noop_with_empty_axes); + ReduceMean_Op(const std::vector<std::int32_t>& axes, bool keep_dims = true, bool noop_with_empty_axes = false); /** * @brief Copy-constructor. Copy the operator attributes and its output tensor(s), but not its input tensors (the new operator has no input associated). @@ -116,8 +116,8 @@ public: // template <DimIdx_t DIM> // const std::string ReduceMean_Op::Type = "ReduceMean"; std::shared_ptr<Node> ReduceMean(const std::vector<std::int32_t> &axes, - bool keep_dims=true, - bool noop_with_empty_axes=false, + bool keep_dims = true, + bool noop_with_empty_axes = false, const std::string& name = ""); } // namespace Aidge