Merge remote-tracking branch 'upstream/master' into add-more-tf-ops-2
This commit is contained in:
commit
d80c1abf5e
@ -32,8 +32,8 @@ end
|
|||||||
|
|
||||||
Base.copy(mx::MXArray) = copy!(Array{Float32}(size(mx)), mx)
|
Base.copy(mx::MXArray) = copy!(Array{Float32}(size(mx)), mx)
|
||||||
|
|
||||||
function MXArray(xs::AbstractArray)
|
function MXArray(xs::AbstractArray, ctx = mx.cpu())
|
||||||
mx = MXArray(size(xs))
|
mx = MXArray(size(xs), ctx)
|
||||||
copy!(mx, xs)
|
copy!(mx, xs)
|
||||||
end
|
end
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user