updated documentation

This commit is contained in:
Garben Tanghe 2020-03-08 14:17:00 +01:00
parent 746e3310f1
commit fc3af681ec
1 changed files with 1 additions and 1 deletions

View File

@ -204,7 +204,7 @@ tversky_loss(ŷ, y; β=eltype(ŷ)(0.7)) = 1 - (sum(y .* ŷ) + 1) / (sum(y .*
""" """
flatten(x::AbstractArray) flatten(x::AbstractArray)
Transforms (w,h,c,b)-shaped input into (w*h*c,b)-shaped output, Transforms (w,h,c,b)-shaped input into (w x h x c,b)-shaped output,
by linearizing all values for each element in the batch. by linearizing all values for each element in the batch.
""" """
function flatten(x::AbstractArray) function flatten(x::AbstractArray)