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

mention treelike

Browse Source
This commit is contained in:
Mike J Innes 2017-11-21 12:59:39 +01:00
parent 187fddc11c
commit e51268caf5
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
docs/src/models/basics.md
View File

@ -151,3 +151,13 @@ m = Chain(x -> x^2, x -> x+1)
m(5) # => 26 m(5) # => 26
``` ```
## Layer helpers
Flux provides a set of helpers for custom layers, which you can enable by calling
```julia
Flux.treelike(Affine)
```
This enables a useful extra set of functionality for our `Affine` layer, such as [collecting its parameters](../training/optimisers.md) or [moving it to the GPU](../gpu.md).