tensor-safe-0.1.0.0: Create valid deep neural network architectures

Safe HaskellNone
LanguageHaskell2010

TensorSafe.Layers.LSTM

Description

This module declares the classing LSTM layer data type.

Synopsis

Documentation

data LSTM :: Nat -> Bool -> Type where Source #

A LSTM layer with a number of units and a option to return the original sequences.

Constructors

LSTM :: LSTM units returnSequences 
Instances
Show (LSTM a b) Source # 
Instance details

Defined in TensorSafe.Layers.LSTM

Methods

showsPrec :: Int -> LSTM a b -> ShowS #

show :: LSTM a b -> String #

showList :: [LSTM a b] -> ShowS #

KnownNat units => Layer (LSTM units b) Source # 
Instance details

Defined in TensorSafe.Layers.LSTM

Methods

layer :: LSTM units b Source #

compile :: LSTM units b -> InputShape -> CNetwork Source #