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

cpu docs

Browse Source
This commit is contained in:
Mike J Innes 2018-03-05 19:25:43 +00:00
parent 6b69edfe6c
commit 5c7f856115
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
docs/src/gpu.md
View File

@ -52,3 +52,19 @@ Tracked 5-element CuArray{Float32,1}:
⋮ ⋮
-0.618002 -0.618002
``` ```
The analogue `cpu` is also available for moving models and data back off of the GPU.
```
julia> x = rand(10) |> gpu
10-element CuArray{Float32,1}:
0.235164
⋮
0.192538
julia> x |> cpu
10-element Array{Float32,1}:
0.235164
⋮
0.192538
```