FluxExtra

Additional layers and functions for the Flux machine learning library.

Functions

Moving models between GPU and CPU

Allows to move branching models between CPU and GPU. Use cpu or gpu as a target.

move(model,target::Union{typeof(cpu),typeof(gpu)})

Layers

Join

Join(dim::Int64)

Concatenates an array of arrays along a dimension dim. A convenient way of using x -> cat(x..., dims = dim).

Split

Split(output::Int64,dim::Int64)

Breaks an array into a number of arrays which is equal to output along a dimension dim.

Addition

Addition()

A convenient way of using x -> sum(x)

Activation

Activation(f::Function)

A convenient way of using x -> some_activation_function.

Identity

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.