Fix ONNX value creation based on FastMatrix (#81)

rwth-i6 · Nov 7, 2024 · 6441be5 · 6441be5
1 parent f0c58ed
commit 6441be5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Onnx/Value.cc b/src/Onnx/Value.cc
@@ -725,7 +725,7 @@ void Value::set(Math::FastMatrix<T> const& mat, bool transpose) {
         // if we transpose we can iterate over both matrices linearly
         for (u32 c = 0u; c < mat.nColumns(); c++) {
             for (u32 r = 0u; r < mat.nRows(); r++) {
-                data[c * mat.nColumns() + r] = mat.at(r, c);
+                data[c * mat.nRows() + r] = mat.at(r, c);
             }
         }
     }