TensorSafe.Layers.LSTM

This module declares the classing LSTM layer data type.

Documentation

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

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)

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)

Defined in TensorSafe.Layers.LSTM

Methods

layer :: LSTM units b

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