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

better mxarray dims api

Browse Source
This commit is contained in:
Mike J Innes 2017-03-08 15:36:38 +00:00
parent 9ecb1311a5
commit 39bb625246
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/backend/mxnet/mxarray.jl
View File

@ -14,7 +14,7 @@ MXArray{T}(data::mx.NDArray, scratch::Array{Float32, T}) = MXArray{T}(data, scra
MXArray(data::mx.NDArray) = MXArray(data, Array{Float32}(size(data)))
MXArray(dims::Integer...) = MXArray(mx.zeros(reverse(dims)))
MXArray(dims::Dims) = MXArray(mx.zeros(reverse(dims)))
Base.size(xs::MXArray) = reverse(size(xs.data))
@ -34,6 +34,6 @@ end
Base.copy(mx::MXArray) = copy!(Array{Float32}(size(mx)), mx)
function MXArray(xs::AbstractArray)
mx = MXArray(size(xs)...)
mx = MXArray(size(xs))
copy!(mx, xs)
end