Port Adjoint, Transpose, and triangular from Nabla

JuliaDiff · Jun 11, 2019 · 51e52e8 · 51e52e8
1 parent 4d46dc7
commit 51e52e8
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/src/rules/linalg/structured.jl b/src/rules/linalg/structured.jl
@@ -16,3 +16,28 @@ rrule(::Type{<:Symmetric}, A::AbstractMatrix) = Symmetric(A), Rule(_symmetric_ba
 
 _symmetric_back(ΔΩ) = UpperTriangular(ΔΩ) + LowerTriangular(ΔΩ)' - Diagonal(ΔΩ)
 _symmetric_back(ΔΩ::Union{Diagonal,UpperTriangular}) = ΔΩ
+
+#####
+##### `Adjoint`
+#####
+
+# TODO: Deal with complex-valued arrays as well
+rrule(::Type{<:Adjoint}, A::AbstractVecOrMat{<:Real}) = Adjoint(A), Rule(adjoint)
+
+rrule(::typeof(adjoint), A::AbstractVecOrMat{<:Real}) = adjoint(A), Rule(adjoint)
+
+#####
+##### `Transpose`
+#####
+
+rrule(::Type{<:Transpose}, A::AbstractVecOrMat) = Transpose(A), Rule(transpose)
+
+rrule(::typeof(transpose), A::AbstractVecOrMat) = transpose(A), Rule(transpose)
+
+#####
+##### Triangular matrices
+#####
+
+rrule(::Type{<:UpperTriangular}, A::AbstractMatrix) = UpperTriangular(A), Rule(Matrix)
+
+rrule(::Type{<:LowerTriangular}, A::AbstractMatrix) = LowerTriangular(A), Rule(Matrix)