From d6f6a15aa2984068bf193499ad85f1c1ac13b8fd Mon Sep 17 00:00:00 2001
From: hrouis <houssemeddine.rouis92@gmail.com>
Date: Mon, 2 Sep 2024 17:08:04 +0200
Subject: [PATCH] fix ReduceMean call
---
src/data/Tensor.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/data/Tensor.cpp b/src/data/Tensor.cpp
index e382fe2ac..abfc91c6c 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());
--
GitLab