[Fix] vit_operators

• Compute axis values in computeOutputDims() instead of forward kernel

include/aidge/operator/ReduceMean.hpp

@@ -79,12 +79,12 @@ class ReduceMean_Op : public OperatorTensor,
                 if (val < 0)
                     val+=static_cast<std::int32_t>(getInput(0)->nbDims());
             });
-            std::sort(axes.cbegin(), axes.cend());
+            std::sort(axes.begin(), axes.end());
 
             // build output dimensions
             std::vector<DimSize_t> outDims = getInput(0)->dims();
             if (this->template getAttr<ReduceMeanAttr::KeepDims>()) {
-                std::for_each(axes.begin(), axes.end(), [&] (const std::int32_t& val) { outDims[val] = 1; });
+                std::for_each(axes.begin(), axes.end(), [&outDims] (const std::int32_t& val) { outDims[val] = 1; });
             }
             else {
                 for (auto it = axes.crbegin(); it != axes.crend(); ++it)