Refactor library structure

2024-03-08 22:20:54 +01:00
parent 1231f4522a
commit 635ef22520
56 changed files with 64 additions and 68 deletions
--- a/bayesnet/utils/bayesnetUtils.h
+++ b/bayesnet/utils/bayesnetUtils.h
@@ -0,0 +1,11 @@
+#ifndef BAYESNET_UTILS_H
+#define BAYESNET_UTILS_H
+#include <vector>
+#include <torch/torch.h>
+namespace bayesnet {
+    std::vector<int> argsort(std::vector<double>& nums);
+    std::vector<std::vector<int>> tensorToVector(torch::Tensor& dtensor);
+    std::vector<std::vector<double>> tensorToVectorDouble(torch::Tensor& dtensor);
+    torch::Tensor vectorToTensor(std::vector<std::vector<int>>& vector, bool transpose = true);
+}
+#endif //BAYESNET_UTILS_H