From 783e6444428ccdb7d903d145071dda1cd0ef1f57 Mon Sep 17 00:00:00 2001 From: Felix Dangel Date: Tue, 18 Jul 2023 09:34:19 -0400 Subject: [PATCH] [FIX] Use column dimension to create random vector --- curvlinops/_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/curvlinops/_base.py b/curvlinops/_base.py index 102faee..f97bc00 100644 --- 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