Torch.NN.Recurrent.Cell.GRU

Documentation

data GRUSpec

Constructors

GRUSpec

Fields

• inputSize :: Int
• hiddenSize :: Int

Instances

Show GRUSpec

Defined in Torch.NN.Recurrent.Cell.GRU

Methods

showsPrec :: Int -> GRUSpec -> ShowS

show :: GRUSpec -> String

showList :: [GRUSpec] -> ShowS

Eq GRUSpec

Defined in Torch.NN.Recurrent.Cell.GRU

Methods

(==) :: GRUSpec -> GRUSpec -> Bool

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

Randomizable GRUSpec GRUCell

Defined in Torch.NN.Recurrent.Cell.GRU

Methods

sample :: GRUSpec -> IO GRUCell

data GRUCell

Constructors

GRUCell

Fields

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

Instances

Generic GRUCell

Defined in Torch.NN.Recurrent.Cell.GRU

Associated Types

type Rep GRUCell

Defined in Torch.NN.Recurrent.Cell.GRU

type Rep GRUCell = D1 ('MetaData "GRUCell" "Torch.NN.Recurrent.Cell.GRU" "hasktorch-0.2.1.2-inplace" 'False) (C1 ('MetaCons "GRUCell" '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 :: GRUCell -> Rep GRUCell x

to :: Rep GRUCell x -> GRUCell

Show GRUCell

Defined in Torch.NN.Recurrent.Cell.GRU

Methods

showsPrec :: Int -> GRUCell -> ShowS

show :: GRUCell -> String

showList :: [GRUCell] -> ShowS

Parameterized GRUCell

Defined in Torch.NN.Recurrent.Cell.GRU

Methods

flattenParameters :: GRUCell -> [Parameter]

_replaceParameters :: GRUCell -> ParamStream GRUCell

Randomizable GRUSpec GRUCell

Defined in Torch.NN.Recurrent.Cell.GRU

Methods

sample :: GRUSpec -> IO GRUCell

type Rep GRUCell

Defined in Torch.NN.Recurrent.Cell.GRU

type Rep GRUCell = D1 ('MetaData "GRUCell" "Torch.NN.Recurrent.Cell.GRU" "hasktorch-0.2.1.2-inplace" 'False) (C1 ('MetaCons "GRUCell" '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))))

gruCellForward

Arguments

:: GRUCell

cell parameters

-> Tensor

input

-> Tensor

hidden

-> Tensor

output