Begin with parameter estimation

test.cc

@@ -1,23 +1,53 @@
-#include <map>
-#include <string>
-#include <iostream>
+// #include <torch/torch.h>
 
-using namespace std;
+// int main()
+// {
+//     torch::Tensor t = torch::rand({ 5, 5 });
 
-int main(int argc, char const* argv[])
+//     // Print original tensor
+//     std::cout << t << std::endl;
+
+//     // New value
+//     torch::Tensor new_val = torch::tensor(10.0f);
+
+//     // Indices for the cell you want to update
+//     auto index_i = torch::tensor({ 2 });
+//     auto index_j = torch::tensor({ 3 });
+
+//     // Update cell
+//     t.index_put_({ index_i, index_j }, new_val);
+
+//     // Print updated tensor
+//     std::cout << t << std::endl;
+// }
+#include <torch/torch.h>
+
+int main()
 {
-    map<string, int> m;
-    m["a"] = 1;
-    m["b"] = 2;
-    m["c"] = 3;
-    if (m.find("b") != m.end()) {
-        cout << "Found b" << endl;
-    } else {
-        cout << "Not found b" << endl;
-    }
-    // for (auto [key, value] : m) {
-    //     cout << key << " " << value << endl;
-    // }
+    torch::Tensor t = torch::rand({ 5, 4, 3 }); // 3D tensor for this example
+    int i = 3, j = 1, k = 2; // Indices for the cell you want to update
+    // Print original tensor
+    std::cout << t << std::endl;
 
-    return 0;
+    // New value
+    torch::Tensor new_val = torch::tensor(10.0f);
+
+    // Indices for the cell you want to update
+    std::vector<torch::Tensor> indices;
+    indices.push_back(torch::tensor(i)); // Replace i with your index for the 1st dimension
+    indices.push_back(torch::tensor(j)); // Replace j with your index for the 2nd dimension
+    indices.push_back(torch::tensor(k)); // Replace k with your index for the 3rd dimension
+    //torch::ArrayRef<at::indexing::TensorIndex> indices_ref(indices);
+    // Update cell
+    //torch::Tensor result = torch::stack(indices);
+    //torch::List<c10::optional<torch::Tensor>> indices_list = { torch::tensor(i), torch::tensor(j), torch::tensor(k) };
+    torch::List<c10::optional<torch::Tensor>> indices_list;
+    indices_list.push_back(torch::tensor(i));
+    indices_list.push_back(torch::tensor(j));
+    indices_list.push_back(torch::tensor(k));
+    //t.index_put_({ torch::tensor(i), torch::tensor(j), torch::tensor(k) }, new_val);
+    t.index_put_(indices_list, new_val);
+
+    // Print updated tensor
+    std::cout << t << std::endl;
 }