diff --git a/src/data/Tensor.cpp b/src/data/Tensor.cpp index e382fe2aca4d6e27a00e4e96233e08b50a92418d..abfc91c6cdf9fd4f6eb46100074b22083514d82e 100644 --- a/src/data/Tensor.cpp +++ b/src/data/Tensor.cpp @@ -125,7 +125,7 @@ Aidge::Tensor Aidge::Tensor::mean() const { // No need to specify the list of all axes! std::vector<std::int32_t> axes(nbDims()); std::iota(std::begin(axes), std::end(axes), 0); - auto mean_ = ReduceMean_Op(axes, 0); + auto mean_ = ReduceMean_Op(axes, false, false); mean_.associateInput(0, std::make_shared<Tensor>(*this)); mean_.setDataType(dataType()); mean_.setDataFormat(dataFormat());