Rebase and add vector normalization

Author: pweglik

packages/react-native-executorch/common/rnexecutorch/data_processing/Numerical.cpp

@@ -1,6 +1,7 @@
 #include "Numerical.h"
 
 #include <algorithm>
+#include <cmath>
 #include <numeric>
 
 namespace rnexecutorch::numerical {
@@ -16,4 +17,15 @@ void softmax(std::vector<float> &v) {
     x /= sum;
   }
 }
+void normalizeVector(std::vector<float> &v) {
+  float norm = 0.0;
+  for (float value : v) {
+    norm += value * value;
+  }
+  norm = sqrt(norm);
+
+  for (float &value : v) {
+    value /= norm;
+  }
+}
 } // namespace rnexecutorch::numerical

packages/react-native-executorch/common/rnexecutorch/data_processing/Numerical.h

@@ -4,4 +4,5 @@
 
 namespace rnexecutorch::numerical {
 void softmax(std::vector<float> &v);
+void normalizeVector(std::vector<float> &v);
 } // namespace rnexecutorch::numerical

packages/react-native-executorch/common/rnexecutorch/models/image_embeddings/ImageEmbeddings.cpp

@@ -1,11 +1,8 @@
 #include "ImageEmbeddings.h"
 
-#include <cstdint>
 #include <executorch/extension/tensor/tensor.h>
-#include <iostream>
-#include <rnexecutorch/Log.h>
 #include <rnexecutorch/data_processing/ImageProcessing.h>
-
+#include <rnexecutorch/data_processing/Numerical.h>
 namespace rnexecutorch {
 
 ImageEmbeddings::ImageEmbeddings(
@@ -48,11 +45,19 @@ ImageEmbeddings::generate(std::string imageSource) {
 
   auto &outputTensor = outputs.at(0).toTensor();
   auto sizesRaw = outputTensor.sizes();
-  std::vector<int32_t> sizes =
-      std::vector<int32_t>(sizesRaw.begin(), sizesRaw.end());
+  auto sizes = std::vector<int32_t>(sizesRaw.begin(), sizesRaw.end());
   size_t bufferSize = outputTensor.numel() * outputTensor.element_size();
   auto buffer = std::make_shared<OwningArrayBuffer>(bufferSize);
-  std::memcpy(buffer->data(), outputTensor.const_data_ptr(), bufferSize);
+
+  std::span<const float> outputTensorSpan(
+      static_cast<const float *>(outputTensor.const_data_ptr()),
+      outputTensor.numel());
+  std::vector<float> outputVector(outputTensorSpan.begin(),
+                                  outputTensorSpan.end());
+
+  numerical::normalizeVector(outputVector);
+
+  std::memcpy(buffer->data(), outputVector.data(), bufferSize);
   auto jsTensor = std::make_shared<JSTensorViewOut>(
       sizes, outputTensor.scalar_type(), buffer);