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

View File

@ -52,3 +52,19 @@ Tracked 5-element CuArray{Float32,1}:
-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
```