fix transpose_banded in scipy.linalg

HIPS · Oct 8, 2023 · 9b5d154 · 9b5d154
1 parent 7f3b929
commit 9b5d154
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autograd/scipy/linalg.py b/autograd/scipy/linalg.py
@@ -52,7 +52,7 @@ def transpose_banded(l_and_u, a):
         T_a = anp.roll(a[:1, :], shifts[0])
         for rr in range(1, num_rows):
             T_a = anp.vstack([T_a, anp.flipud(anp.roll(a[rr:rr+1, :], shifts[rr]))])
-            T_a = anp.flipud(T_a)
+        T_a = anp.flipud(T_a)
 
         T_l_and_u = anp.flip(l_and_u)