hasktorch-0.2.1.6: Haskell bindings to libtorch, supporting both typed and untyped tensors.
Safe HaskellNone
LanguageHaskell2010

Torch.NN.Recurrent.Cell.LSTM

Documentation

data LSTMSpec Source #

Constructors

LSTMSpec 

Fields

Instances

Instances details
Show LSTMSpec Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

showsPrec :: Int -> LSTMSpec -> ShowS #

show :: LSTMSpec -> String #

showList :: [LSTMSpec] -> ShowS #

Eq LSTMSpec Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

(==) :: LSTMSpec -> LSTMSpec -> Bool #

(/=) :: LSTMSpec -> LSTMSpec -> Bool #

Randomizable LSTMSpec LSTMCell Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

sample :: LSTMSpec -> IO LSTMCell Source #

data LSTMCell Source #

Constructors

LSTMCell 

Fields

Instances

Instances details
Generic LSTMCell Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.LSTM

Associated Types

type Rep LSTMCell 
Instance details

Defined in Torch.NN.Recurrent.Cell.LSTM

type Rep LSTMCell = D1 ('MetaData "LSTMCell" "Torch.NN.Recurrent.Cell.LSTM" "hasktorch-0.2.1.6-LDjqZXcNPYQAOVyrR3h8rq" 'False) (C1 ('MetaCons "LSTMCell" 'PrefixI 'True) ((S1 ('MetaSel ('Just "weightsIH") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "weightsHH") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter)) :*: (S1 ('MetaSel ('Just "biasIH") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "biasHH") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter))))

Methods

from :: LSTMCell -> Rep LSTMCell x #

to :: Rep LSTMCell x -> LSTMCell #

Show LSTMCell Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

showsPrec :: Int -> LSTMCell -> ShowS #

show :: LSTMCell -> String #

showList :: [LSTMCell] -> ShowS #

Parameterized LSTMCell Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

flattenParameters :: LSTMCell -> [Parameter] Source #

_replaceParameters :: LSTMCell -> ParamStream LSTMCell Source #

Randomizable LSTMSpec LSTMCell Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

sample :: LSTMSpec -> IO LSTMCell Source #

type Rep LSTMCell Source # 
Instance details

Defined in Torch.NN.Recurrent.Cell.LSTM

type Rep LSTMCell = D1 ('MetaData "LSTMCell" "Torch.NN.Recurrent.Cell.LSTM" "hasktorch-0.2.1.6-LDjqZXcNPYQAOVyrR3h8rq" 'False) (C1 ('MetaCons "LSTMCell" 'PrefixI 'True) ((S1 ('MetaSel ('Just "weightsIH") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "weightsHH") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter)) :*: (S1 ('MetaSel ('Just "biasIH") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter) :*: S1 ('MetaSel ('Just "biasHH") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 Parameter))))

lstmCellForward Source #

Arguments

:: LSTMCell

cell parameters

-> (Tensor, Tensor)

(hidden, cell)

-> Tensor

input

-> (Tensor, Tensor)

output (hidden, cell)