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

enable nested broadcast

Browse Source
This commit is contained in:
Mike J Innes 2018-09-07 02:05:03 +01:00
parent 395a35d137
commit 6bbed07e96
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

@ -351,9 +351,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)