Update src/rulesets/LinearAlgebra/structured.jl

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