rmontanana/Platform
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix tensor utils ambiguous call

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2025-07-21 11:40:16 +02:00
parent b990684581
commit f68d216150
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/common/TensorUtils.hpp
View File

@@ -62,7 +62,7 @@ namespace platform {
torch::Tensor tensor = torch::empty({ static_cast<long>(rows), static_cast<long>(cols) }, torch::kInt64); 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 i = 0; i < rows; ++i) {
for (size_t j = 0; j < cols; ++j) { 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; return tensor;