Fix problems due to rebase
This commit is contained in:
parent
e16c24a9b8
commit
64ce32ddcf
|
@ -17,7 +17,6 @@ makedocs(modules=[Flux, NNlib],
|
|||
"DataLoader" => "data/dataloader.md"],
|
||||
"Training Models" =>
|
||||
["Optimisers" => "training/optimisers.md",
|
||||
"Loss Functions" => "training/loss_functions.md",
|
||||
"Training" => "training/training.md"],
|
||||
"GPU Support" => "gpu.md",
|
||||
"Saving & Loading" => "saving.md",
|
||||
|
|
|
@ -1,13 +0,0 @@
|
|||
# Loss Functions
|
||||
|
||||
The following functions provide basic loss (or cost) functions.
|
||||
|
||||
```@docs
|
||||
Flux.mse
|
||||
Flux.crossentropy
|
||||
Flux.logitcrossentropy
|
||||
Flux.binarycrossentropy
|
||||
Flux.logitbinarycrossentropy
|
||||
Flux.normalise
|
||||
```
|
||||
|
|
@ -98,10 +98,9 @@ julia> d = Dense(5, 2)
|
|||
Dense(5, 2)
|
||||
|
||||
julia> d(rand(5))
|
||||
Array{Float64,1}:
|
||||
0.00257447
|
||||
-0.00449443
|
||||
```
|
||||
2-element Array{Float32,1}:
|
||||
-0.16210233
|
||||
0.12311903```
|
||||
"""
|
||||
struct Dense{F,S,T}
|
||||
W::S
|
||||
|
|
|
@ -128,7 +128,13 @@ Split `xs` into `n` parts.
|
|||
# Examples
|
||||
```jldoctest
|
||||
julia> Flux.chunk(1:10, 3)
|
||||
3-element Array{Array{Int64,1},1}:
|
||||
3-element Array{UnitRange{Int64},1}:
|
||||
1:4
|
||||
5:8
|
||||
9:10
|
||||
|
||||
julia> Flux.chunk(collect(1:10), 3)
|
||||
3-element Array{SubArray{Int64,1,Array{Int64,1},Tuple{UnitRange{Int64}},true},1}:
|
||||
[1, 2, 3, 4]
|
||||
[5, 6, 7, 8]
|
||||
[9, 10]
|
||||
|
|
Loading…
Reference in New Issue