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

julia's map is shitty

Browse Source
This commit is contained in:
Mike J Innes 2017-02-21 09:05:08 +00:00
parent 521258eb08
commit 417a70713b
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/backend/mxnet/model.jl
View File

@ -13,6 +13,7 @@ tond(xs::AArray) = copy!(mx.zeros(size(xs)), xs)
ndzero!(xs::mx.NDArray) = copy!(xs, mx.zeros(size(xs))) ndzero!(xs::mx.NDArray) = copy!(xs, mx.zeros(size(xs)))
function mxargs(args) function mxargs(args)
isempty(args) && return Dict{Symbol,mx.NDArray}()
map(args) do kv map(args) do kv
arg, value = kv arg, value = kv
arg => tond(value) arg => tond(value)
@ -20,6 +21,7 @@ function mxargs(args)
end end
function mxgrads(mxargs) function mxgrads(mxargs)
isempty(mxargs) && return Dict{Symbol,mx.NDArray}()
map(mxargs) do kv map(mxargs) do kv
arg, value = kv arg, value = kv
arg => mx.zeros(size(value)) arg => mx.zeros(size(value))