LSTM: Create an LSTM layer with 'in_size' input size, and...

In enweg/BFluxR: Implementation of Bayesian Neural Networks

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)

Arguments

in_size	Input size
out_size	Output size

Value

A list with the following content

• in_size - Input Size

• out_size - Output Size

• julia - Julia code representing the Layer

See Also

Dense

Examples

## Not run: 
  ## 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))

## End(Not run)

enweg/BFluxR documentation built on Jan. 27, 2024, 6:43 p.m.