Add roc-auc-ovr as score to b_main

src/common/Utils.h

@@ -4,7 +4,17 @@
 #include <string>
 #include <vector>
 #include <algorithm>
+#include <torch/torch.h>
 namespace platform {
+    template <typename T>
+    std::vector<T> tensorToVector(const torch::Tensor& tensor)
+    {
+        torch::Tensor contig_tensor = tensor.contiguous();
+        auto num_elements = contig_tensor.numel();
+        const T* tensor_data = contig_tensor.data_ptr<T>();
+        std::vector<T> result(tensor_data, tensor_data + num_elements);
+        return result;
+    }
     static std::string trim(const std::string& str)
     {
         std::string result = str;