[FIX] Use column dimension to create random vector

f-dangel · Jul 18, 2023 · 783e644 · 783e644
1 parent f3d4bd9
commit 783e644
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/curvlinops/_base.py b/curvlinops/_base.py
@@ -154,7 +154,7 @@ def _check_deterministic(self):
             self.print_nonclose(grad1, grad2, rtol, atol)
             raise RuntimeError("Check for deterministic gradient failed.")
 
-        v = rand(self.shape[0]).astype(self.dtype)
+        v = rand(self.shape[1]).astype(self.dtype)
         mat_v1 = self @ v
         mat_v2 = self @ v