Normalizations

FluxExtra.JoinType
Join(dim::Int64)
Join(dim = dim::Int64)

Concatenates a tuple of arrays along a dimension dim. A convenient and type stable way of using x -> cat(x..., dims = dim).

FluxExtra.SplitType
Split(outputs::Int64, dim::Int64)
Split(outputs::Int64, dim = dim::Int64)

Breaks an array into a number of arrays which is equal to outputs along a dimension dim. dim should we divisible by outputs without a remainder.

FluxExtra.FlattenType
Flatten()

Flattens an array. A convenient way of using x -> Flux.flatten(x).

FluxExtra.IdentityType
Identity()

Returns its input without changes. Should be used with a Parallel layer if one wants to have a branch that does not change its input.