BayesFluxR (version 0.1.3)

LSTM: Create an LSTM layer with `in_size` input size, and `out_size` hidden state size

Description

Create an LSTM layer with `in_size` input size, and `out_size` hidden state size

Usage

LSTM(in_size, out_size)

Value

A list with the following content

• in_size - Input Size

• out_size - Output Size

• julia - Julia code representing the Layer

Arguments

in_size

Input size

out_size

Output size

See Also

Dense

Examples

if (FALSE) {
  ## Needs previous call to `BayesFluxR_setup` which is time
  ## consuming and requires Julia and BayesFlux.jl
  BayesFluxR_setup(installJulia=TRUE, seed=123)
  net <- Chain(LSTM(5, 5))
}