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

fix ReduceMean.hpp initializer_list as vector

parent cfd265eb
No related branches found
No related tags found
No related merge requests found
......@@ -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}));
}
}
......
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