TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed issue #542.

Browse Source 
Added tracking of LinearAlgebra.det and its grad method.
This commit is contained in:
Ivan Yashchuk 2019-02-05 11:38:27 +02:00
parent 940b1e6dbf
commit aa64d2157d
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/tracker/lib/array.jl
View File

@ -1,7 +1,7 @@
import Base: *
import LinearAlgebra
import LinearAlgebra: inv, \, /
import LinearAlgebra: inv, det, \, /
using Statistics
using LinearAlgebra: Transpose, Adjoint, diagm, diag
@ -124,6 +124,9 @@ Base.adjoint(xs::TrackedArray) = track(adjoint, xs)
@grad transpose(xs) = transpose(data(xs)), Δ -> (trim(xs, transpose(Δ)),)
@grad adjoint(xs) = data(xs)', Δ -> (trim(xs, Δ'),)
det(xs::TrackedArray) = track(det, xs)
@grad det(xs) = det(data(xs)), Δ -> (Δ * transpose(adjoint(xs)),)
Base.repeat(xs::TrackedArray; kw...) = track(repeat, xs; kw...)
@grad function repeat(xs; inner=ntuple(x->1, ndims(xs)), outer=ntuple(x->1, ndims(xs)))