From a3cda9016c48ee19367e7732bdb560b90ef5fe5b Mon Sep 17 00:00:00 2001 From: Roger-luo Date: Thu, 25 Oct 2018 13:48:33 -0400 Subject: [PATCH] apply Mike's change --- src/tracker/array.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/tracker/array.jl b/src/tracker/array.jl index f13feb77..9c89b5f6 100644 --- 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