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

Use other definition for grad(det(A)).

Browse Source
This commit is contained in:
Ivan Yashchuk 2019-02-05 14:36:28 +02:00
parent aa64d2157d
commit f790fff59a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -125,7 +125,7 @@ Base.adjoint(xs::TrackedArray) = track(adjoint, xs)
@grad adjoint(xs) = data(xs)', Δ -> (trim(xs, Δ'),)
det(xs::TrackedArray) = track(det, xs)
@grad det(xs) = det(data(xs)), Δ -> (Δ * transpose(adjoint(xs)),)
@grad det(xs) = det(data(xs)), Δ -> (Δ * det(xs) * transpose(inv(xs)),)
Base.repeat(xs::TrackedArray; kw...) = track(repeat, xs; kw...)