enable nested broadcast

2018-09-07 02:05:03 +01:00 · 2018-09-07 02:05:03 +01:00 · 6bbed07e96
parent 395a35d137
commit 6bbed07e96
1 changed files with 3 additions and 3 deletions
--- a/src/tracker/array.jl
+++ b/src/tracker/array.jl
@ -351,9 +351,9 @@ end
  eltype(y) <: Real || return y
  eltype(y) == Bool && return y
  function back(Δ)
-    Δargs = ntuple(i -> partial.(f, data(Δ), i, args...), Val(N))
-    dxs = unbroadcast.(args, Δargs)
-    return nobacksies(:broadcast, dxs)
+    Δargs = ntuple(i -> partial.(f, Δ, i, args...), Val(N))
+    dxs = map(unbroadcast, args, Δargs)
+    return dxs
  end
  # So we can return non-tracked arrays
  track(Call(back, tracker.(args)), y)