Update src/rulesets/LinearAlgebra/structured.jl

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
JuliaDiff · Nov 22, 2023 · a275f1f · a275f1f
1 parent 3459acb
commit a275f1f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/rulesets/LinearAlgebra/structured.jl b/src/rulesets/LinearAlgebra/structured.jl
@@ -276,8 +276,12 @@ end
 function rrule(::Type{Tridiagonal}, dl, d, du)
     y = Tridiagonal(dl, d, du)
     @views function ∇Tridiagonal(∂y)
-        return (NoTangent(), diag(∂y[2:end, 1:(end - 1)]), diag(∂y),
-            diag(∂y[1:(end - 1), 2:end]))
+        return (
+            NoTangent(),
+            diag(∂y[2:end, 1:(end - 1)]),
+            diag(∂y),
+            diag(∂y[1:(end - 1), 2:end]),
+        )
     end
     return y, ∇Tridiagonal
 end