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

fixes #403

Browse Source
This commit is contained in:
Mike J Innes 2018-10-10 12:26:03 +01:00
parent 3285afa45a
commit 9f6c3d5a2c
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/tracker/array.jl
View File

@ -376,8 +376,7 @@ unbroadcast(x::AbstractArray, Δ) =
trim(x, sum(Δ, dims = ntuple(i -> size(x, i) == 1 ? i : ndims(Δ)+1, Val(ndims(Δ))))) trim(x, sum(Δ, dims = ntuple(i -> size(x, i) == 1 ? i : ndims(Δ)+1, Val(ndims(Δ)))))
unbroadcast(x::Number, Δ) = sum(Δ) unbroadcast(x::Number, Δ) = sum(Δ)
unbroadcast(x::Base.RefValue{<:Function}, _) = nothing unbroadcast(x::Base.RefValue, _) = nothing
unbroadcast(x::Base.RefValue{<:Val}, _) = nothing
dual(x, p) = x dual(x, p) = x
dual(x::Real, p) = Dual(x, p) dual(x::Real, p) = Dual(x, p)