diff --git a/include/aidge/operator/ReduceMean.hpp b/include/aidge/operator/ReduceMean.hpp index 609f5be5f28efee8ebd266b6756b5bf172e45bd5..5c5531f343661f3e536d2519615e030c559a8256 100644 --- a/include/aidge/operator/ReduceMean.hpp +++ b/include/aidge/operator/ReduceMean.hpp @@ -96,7 +96,7 @@ class ReduceMean_Op : public OperatorTensor, outDims.erase(outDims.begin() + static_cast<std::size_t>(*it)); } - mOutputs[0]->resize((outDims.size()>0) ? outDims : {1}); + mOutputs[0]->resize((outDims.size()>0) ? outDims : std::vector<DimSize_t>({1})); } }