Fix tensor utils ambiguous call

2025-07-21 11:40:16 +02:00
parent b990684581
commit f68d216150
1 changed files with 1 additions and 1 deletions
--- 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;