From f68d216150bdede0d8c3ea019a6098d21e80e2cc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= <rmontanana@gmail.com>
Date: Mon, 21 Jul 2025 11:40:16 +0200
Subject: [PATCH] Fix tensor utils ambiguous call

---
 src/common/TensorUtils.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/common/TensorUtils.hpp b/src/common/TensorUtils.hpp
index eb69371..93b258e 100644
--- a/src/common/TensorUtils.hpp
+++ b/src/common/TensorUtils.hpp
@@ -62,7 +62,7 @@ namespace platform {
             torch::Tensor tensor = torch::empty({ static_cast<long>(rows), static_cast<long>(cols) }, torch::kInt64);
             for (size_t i = 0; i < rows; ++i) {
                 for (size_t j = 0; j < cols; ++j) {
-                    tensor.index_put_({ static_cast<long>(i), static_cast<long>(j) }, data[i][j]);
+                    tensor.index_put_({static_cast<int64_t>(i), static_cast<int64_t>(j)}, torch::scalar_tensor(data[i][j]));
                 }
             }
             return tensor;