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

apply Mike's change

Browse Source
This commit is contained in:
Roger-luo 2018-10-25 13:48:33 -04:00
parent 5f99e5775a
commit a3cda9016c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/tracker/array.jl
View File

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