| Safe Haskell | None | 
|---|---|
| Language | Haskell2010 | 
Backprop.Learn.Model.Neural.LSTM
Synopsis
- lstm :: (KnownNat i, KnownNat o) => Model (Just (LSTMp i o)) (Just (R o :# R o)) (R i) (R o)
- data LSTMp (i :: Nat) (o :: Nat) = LSTMp {- _lstmForget :: !(FCp (i + o) o)
- _lstmInput :: !(FCp (i + o) o)
- _lstmUpdate :: !(FCp (i + o) o)
- _lstmOutput :: !(FCp (i + o) o)
 
- lstmForget :: forall i o. Lens' (LSTMp i o) (FCp ((+) i o) o)
- lstmInput :: forall i o. Lens' (LSTMp i o) (FCp ((+) i o) o)
- lstmUpdate :: forall i o. Lens' (LSTMp i o) (FCp ((+) i o) o)
- lstmOutput :: forall i o. Lens' (LSTMp i o) (FCp ((+) i o) o)
- reshapeLSTMpInput :: (ContGen d, PrimMonad m, KnownNat i, KnownNat i', KnownNat o) => d -> Gen (PrimState m) -> LSTMp i o -> m (LSTMp i' o)
- reshapeLSTMpOutput :: (ContGen d, PrimMonad m, KnownNat i, KnownNat o, KnownNat o') => d -> Gen (PrimState m) -> LSTMp i o -> m (LSTMp i o')
- lstm' :: (KnownNat i, KnownNat o) => Model (Just (LSTMp i o)) (Just (R o)) (R (i + o)) (R o)
- gru :: (KnownNat i, KnownNat o) => Model (Just (GRUp i o)) (Just (R o)) (R i) (R o)
- data GRUp (i :: Nat) (o :: Nat) = GRUp {- _gruMemory :: !(FCp (i + o) o)
- _gruUpdate :: !(FCp (i + o) o)
- _gruOutput :: !(FCp (i + o) o)
 
- gruMemory :: forall i o. Lens' (GRUp i o) (FCp ((+) i o) o)
- gruUpdate :: forall i o. Lens' (GRUp i o) (FCp ((+) i o) o)
- gruOutput :: forall i o. Lens' (GRUp i o) (FCp ((+) i o) o)
- gru' :: forall i o. (KnownNat i, KnownNat o) => Model (Just (GRUp i o)) Nothing (R (i + o)) (R o)
LSTM
lstm :: (KnownNat i, KnownNat o) => Model (Just (LSTMp i o)) (Just (R o :# R o)) (R i) (R o) Source #
Long-term short-term memory layer
data LSTMp (i :: Nat) (o :: Nat) Source #
LSTM layer parmateters
Constructors
| LSTMp | |
| Fields 
 | |
Instances
reshapeLSTMpInput :: (ContGen d, PrimMonad m, KnownNat i, KnownNat i', KnownNat o) => d -> Gen (PrimState m) -> LSTMp i o -> m (LSTMp i' o) Source #
reshapeLSTMpOutput :: (ContGen d, PrimMonad m, KnownNat i, KnownNat o, KnownNat o') => d -> Gen (PrimState m) -> LSTMp i o -> m (LSTMp i o') Source #
lstm' :: (KnownNat i, KnownNat o) => Model (Just (LSTMp i o)) (Just (R o)) (R (i + o)) (R o) Source #
Stateless version of lstm that takes the "previous input" as a part
 of the input vector.
GRU
gru :: (KnownNat i, KnownNat o) => Model (Just (GRUp i o)) (Just (R o)) (R i) (R o) Source #
Gated Recurrent Unit
data GRUp (i :: Nat) (o :: Nat) Source #
GRU layer parmateters
Constructors
| GRUp | |
| Fields 
 | |