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

Add new-style diagm to tracker

Browse Source
This commit is contained in:
Eric Davies 2018-10-26 13:39:49 -05:00
parent 44ccdb7ca9
commit 9f9803eec6
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/tracker/array.jl
View File

@ -309,6 +309,9 @@ end
# BLAS # BLAS
LinearAlgebra.diagm(x::Pair{<:Integer, <:TrackedVector}) = track(diagm, x)
@grad diagm(x::Pair) = diagm(x[1] => data(x[2])), Δ -> (diag(Δ, x[1]),)
LinearAlgebra.diagm(x::TrackedVector) = track(diagm, x) LinearAlgebra.diagm(x::TrackedVector) = track(diagm, x)
@grad diagm(x) = diagm(data(x)), Δ -> (diag(Δ),) @grad diagm(x) = diagm(data(x)), Δ -> (diag(Δ),)