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

Merge pull request #397 from FluxML/nest-bcast

Browse Source 
Nested Derivatives of Broadcast
This commit is contained in:
Mike J Innes 2018-09-14 20:23:28 +01:00 committed by GitHub
commit 08fb9b7df1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/tracker/array.jl
View File

@ -353,9 +353,9 @@ end
eltype(y) <: Real || return y eltype(y) <: Real || return y
eltype(y) == Bool && return y eltype(y) == Bool && return y
function back(Δ) function back(Δ)
Δargs = ntuple(i -> partial.(f, data(Δ), i, args...), Val(N)) Δargs = ntuple(i -> partial.(f, Δ, i, args...), Val(N))
dxs = unbroadcast.(args, Δargs) dxs = map(unbroadcast, args, Δargs)
return nobacksies(:broadcast, dxs) return dxs
end end
# So we can return non-tracked arrays # So we can return non-tracked arrays
track(Call(back, tracker.(args)), y) track(Call(back, tracker.(args)), y)