Torch.NN.Recurrent.Cell.LSTM

Documentation

data LSTMSpec

Constructors

LSTMSpec

Fields

• inputSize :: Int
• hiddenSize :: Int

Instances

Show LSTMSpec

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

showsPrec :: Int -> LSTMSpec -> ShowS

show :: LSTMSpec -> String

showList :: [LSTMSpec] -> ShowS

Eq LSTMSpec

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

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

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

Randomizable LSTMSpec LSTMCell

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

sample :: LSTMSpec -> IO LSTMCell

data LSTMCell

Constructors

LSTMCell

Fields

• weightsIH :: Parameter
• weightsHH :: Parameter
• biasIH :: Parameter
• biasHH :: Parameter

Instances

Generic LSTMCell

Defined in Torch.NN.Recurrent.Cell.LSTM

Associated Types

type Rep LSTMCell :: Type -> Type

Methods

from :: LSTMCell -> Rep LSTMCell x

to :: Rep LSTMCell x -> LSTMCell

Show LSTMCell

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

showsPrec :: Int -> LSTMCell -> ShowS

show :: LSTMCell -> String

showList :: [LSTMCell] -> ShowS

Parameterized LSTMCell

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

flattenParameters :: LSTMCell -> [Parameter]

_replaceParameters :: LSTMCell -> ParamStream LSTMCell

Randomizable LSTMSpec LSTMCell

Defined in Torch.NN.Recurrent.Cell.LSTM

Methods

sample :: LSTMSpec -> IO LSTMCell

type Rep LSTMCell

Defined in Torch.NN.Recurrent.Cell.LSTM

type Rep LSTMCell = D1 ('MetaData "LSTMCell" "Torch.NN.Recurrent.Cell.LSTM" "hasktorch-0.2.1.1-inplace" '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

Arguments

:: LSTMCell

cell parameters

-> (Tensor, Tensor)

(hidden, cell)

-> Tensor

input

-> (Tensor, Tensor)

output (hidden, cell)