apply Mike's change

2018-10-25 13:48:33 -04:00 · 2018-10-25 13:48:33 -04:00 · a3cda9016c
commit a3cda9016c
parent 5f99e5775a
1 changed files with 2 additions and 2 deletions
--- a/src/tracker/array.jl
+++ b/src/tracker/array.jl
@ -86,9 +86,9 @@ Base.view(x::TrackedArray, inds...) = track(Base.view, x, inds...)

@grad function view(x::AbstractArray, inds...)
    view(data(x), inds...), function (Δ)
-        grad_output = fill!(similar(data(x)), 0)
+        grad_output = zero(x)
        subgrad = view(grad_output, inds...)
-        setindex!(subgrad, Δ, :)
+        subgrad[:] = Δ
        (grad_output, map(_->nothing, inds)...)
    end
 end