Safe Haskell | None |
---|---|
Language | Haskell2010 |
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
LSTMp | |
|
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
GRUp | |
|