Safe Haskell	None
Language	Haskell2010

Torch.Typed.NN.Recurrent.Cell.GRU

Synopsis

data GRUCellSpec (inputDim :: Nat) (hiddenDim :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) = GRUCellSpec
data GRUCell (inputDim :: Nat) (hiddenDim :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) = GRUCell {
- gruCell_w_ih :: Parameter device dtype '[3 * hiddenDim, inputDim]
- gruCell_w_hh :: Parameter device dtype '[3 * hiddenDim, hiddenDim]
- gruCell_b_ih :: Parameter device dtype '[3 * hiddenDim]
- gruCell_b_hh :: Parameter device dtype '[3 * hiddenDim]
}
gruCellForward :: forall (inputDim :: Nat) (hiddenDim :: Nat) (batchSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)). (KnownDType dtype, KnownNat inputDim, KnownNat hiddenDim, KnownNat batchSize) => GRUCell inputDim hiddenDim dtype device -> Tensor device dtype '[batchSize, hiddenDim] -> Tensor device dtype '[batchSize, inputDim] -> Tensor device dtype '[batchSize, hiddenDim]
gruFold :: forall (inputDim :: Nat) (hiddenDim :: Nat) (batchSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)). (KnownDType dtype, KnownNat inputDim, KnownNat hiddenDim, KnownNat batchSize) => GRUCell inputDim hiddenDim dtype device -> Tensor device dtype '[batchSize, hiddenDim] -> [Tensor device dtype '[batchSize, inputDim]] -> Tensor device dtype '[batchSize, hiddenDim]
gruCellScan :: forall (inputDim :: Nat) (hiddenDim :: Nat) (batchSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)). (KnownDType dtype, KnownNat inputDim, KnownNat hiddenDim, KnownNat batchSize) => GRUCell inputDim hiddenDim dtype device -> Tensor device dtype '[batchSize, hiddenDim] -> [Tensor device dtype '[batchSize, inputDim]] -> [Tensor device dtype '[batchSize, hiddenDim]]

Documentation

data GRUCellSpec (inputDim :: Nat) (hiddenDim :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) Source #

A specification for a gated recurrent unit (GRU) cell.

Constructors

GRUCellSpec

Weights and biases are drawn from the standard normal distibution (having mean 0 and variance 1)

Instances

Instances details

Bounded (GRUCellSpec inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Methods

minBound :: GRUCellSpec inputDim hiddenDim dtype device #

maxBound :: GRUCellSpec inputDim hiddenDim dtype device #

Enum (GRUCellSpec inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Methods

succ :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device #

pred :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device #

toEnum :: Int -> GRUCellSpec inputDim hiddenDim dtype device #

fromEnum :: GRUCellSpec inputDim hiddenDim dtype device -> Int #

enumFrom :: GRUCellSpec inputDim hiddenDim dtype device -> [GRUCellSpec inputDim hiddenDim dtype device] #

enumFromThen :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> [GRUCellSpec inputDim hiddenDim dtype device] #

enumFromTo :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> [GRUCellSpec inputDim hiddenDim dtype device] #

enumFromThenTo :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> [GRUCellSpec inputDim hiddenDim dtype device] #

Generic (GRUCellSpec inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Associated Types

type Rep (GRUCellSpec inputDim hiddenDim dtype device)
Instance details Defined in Torch.Typed.NN.Recurrent.Cell.GRU type Rep (GRUCellSpec inputDim hiddenDim dtype device) = D1 ('MetaData "GRUCellSpec" "Torch.Typed.NN.Recurrent.Cell.GRU" "hasktorch-0.2.1.2-inplace" 'False) (C1 ('MetaCons "GRUCellSpec" 'PrefixI 'False) (U1 :: Type -> Type))

Methods

from :: GRUCellSpec inputDim hiddenDim dtype device -> Rep (GRUCellSpec inputDim hiddenDim dtype device) x #

to :: Rep (GRUCellSpec inputDim hiddenDim dtype device) x -> GRUCellSpec inputDim hiddenDim dtype device #

Show (GRUCellSpec inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Methods

showsPrec :: Int -> GRUCellSpec inputDim hiddenDim dtype device -> ShowS #

show :: GRUCellSpec inputDim hiddenDim dtype device -> String #

showList :: [GRUCellSpec inputDim hiddenDim dtype device] -> ShowS #

Eq (GRUCellSpec inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Methods

(==) :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> Bool #

(/=) :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> Bool #

Ord (GRUCellSpec inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Methods

compare :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> Ordering #

(<) :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> Bool #

(<=) :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> Bool #

(>) :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> Bool #

(>=) :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> Bool #

max :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device #

min :: GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device -> GRUCellSpec inputDim hiddenDim dtype device #

(KnownDevice device, KnownDType dtype, KnownNat inputDim, KnownNat hiddenDim, RandDTypeIsValid device dtype) => Randomizable (GRUCellSpec inputDim hiddenDim dtype device) (GRUCell inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Methods

sample :: GRUCellSpec inputDim hiddenDim dtype device -> IO (GRUCell inputDim hiddenDim dtype device) Source #

type Rep (GRUCellSpec inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

type Rep (GRUCellSpec inputDim hiddenDim dtype device) = D1 ('MetaData "GRUCellSpec" "Torch.Typed.NN.Recurrent.Cell.GRU" "hasktorch-0.2.1.2-inplace" 'False) (C1 ('MetaCons "GRUCellSpec" 'PrefixI 'False) (U1 :: Type -> Type))

data GRUCell (inputDim :: Nat) (hiddenDim :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)) Source #

A gated recurrent unit (GRU) cell.

Constructors

GRUCell
Fields gruCell_w_ih :: Parameter device dtype '[3 * hiddenDim, inputDim] input-to-hidden weights gruCell_w_hh :: Parameter device dtype '[3 * hiddenDim, hiddenDim] hidden-to-hidden weights gruCell_b_ih :: Parameter device dtype '[3 * hiddenDim] input-to-hidden bias gruCell_b_hh :: Parameter device dtype '[3 * hiddenDim] hidden-to-hidden bias

Instances

Instances details

Generic (GRUCell inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Associated Types

type Rep (GRUCell inputDim hiddenDim dtype device)

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

type Rep (GRUCell inputDim hiddenDim dtype device) = D1 ('MetaData "GRUCell" "Torch.Typed.NN.Recurrent.Cell.GRU" "hasktorch-0.2.1.2-inplace" 'False) (C1 ('MetaCons "GRUCell" 'PrefixI 'True) ((S1 ('MetaSel ('Just "gruCell_w_ih") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[3 * hiddenDim, inputDim])) :*: S1 ('MetaSel ('Just "gruCell_w_hh") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[3 * hiddenDim, hiddenDim]))) :*: (S1 ('MetaSel ('Just "gruCell_b_ih") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[3 * hiddenDim])) :*: S1 ('MetaSel ('Just "gruCell_b_hh") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[3 * hiddenDim])))))

Methods

from :: GRUCell inputDim hiddenDim dtype device -> Rep (GRUCell inputDim hiddenDim dtype device) x #

to :: Rep (GRUCell inputDim hiddenDim dtype device) x -> GRUCell inputDim hiddenDim dtype device #

Show (GRUCell inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Methods

showsPrec :: Int -> GRUCell inputDim hiddenDim dtype device -> ShowS #

show :: GRUCell inputDim hiddenDim dtype device -> String #

showList :: [GRUCell inputDim hiddenDim dtype device] -> ShowS #

Parameterized (GRUCell inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Associated Types

type Parameters (GRUCell inputDim hiddenDim dtype device)
Instance details Defined in Torch.Typed.NN.Recurrent.Cell.GRU type Parameters (GRUCell inputDim hiddenDim dtype device) = GParameters (Rep (GRUCell inputDim hiddenDim dtype device))

Methods

flattenParameters :: GRUCell inputDim hiddenDim dtype device -> HList (Parameters (GRUCell inputDim hiddenDim dtype device)) Source #

replaceParameters :: GRUCell inputDim hiddenDim dtype device -> HList (Parameters (GRUCell inputDim hiddenDim dtype device)) -> GRUCell inputDim hiddenDim dtype device Source #

(KnownDevice device, KnownDType dtype, KnownNat inputDim, KnownNat hiddenDim, RandDTypeIsValid device dtype) => Randomizable (GRUCellSpec inputDim hiddenDim dtype device) (GRUCell inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

Methods

sample :: GRUCellSpec inputDim hiddenDim dtype device -> IO (GRUCell inputDim hiddenDim dtype device) Source #

type Rep (GRUCell inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

type Rep (GRUCell inputDim hiddenDim dtype device) = D1 ('MetaData "GRUCell" "Torch.Typed.NN.Recurrent.Cell.GRU" "hasktorch-0.2.1.2-inplace" 'False) (C1 ('MetaCons "GRUCell" 'PrefixI 'True) ((S1 ('MetaSel ('Just "gruCell_w_ih") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[3 * hiddenDim, inputDim])) :*: S1 ('MetaSel ('Just "gruCell_w_hh") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[3 * hiddenDim, hiddenDim]))) :*: (S1 ('MetaSel ('Just "gruCell_b_ih") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[3 * hiddenDim])) :*: S1 ('MetaSel ('Just "gruCell_b_hh") 'NoSourceUnpackedness 'NoSourceStrictness 'DecidedStrict) (Rec0 (Parameter device dtype '[3 * hiddenDim])))))

type Parameters (GRUCell inputDim hiddenDim dtype device) Source #

Instance details

Defined in Torch.Typed.NN.Recurrent.Cell.GRU

type Parameters (GRUCell inputDim hiddenDim dtype device) = GParameters (Rep (GRUCell inputDim hiddenDim dtype device))

gruCellForward Source #

Arguments

:: forall (inputDim :: Nat) (hiddenDim :: Nat) (batchSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)). (KnownDType dtype, KnownNat inputDim, KnownNat hiddenDim, KnownNat batchSize)
=> GRUCell inputDim hiddenDim dtype device	The cell
-> Tensor device dtype '[batchSize, hiddenDim]	The current Hidden state
-> Tensor device dtype '[batchSize, inputDim]	The input
-> Tensor device dtype '[batchSize, hiddenDim]	The subsequent Hidden state

A single recurrent step of a GRUCell

gruFold Source #

Arguments

:: forall (inputDim :: Nat) (hiddenDim :: Nat) (batchSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)). (KnownDType dtype, KnownNat inputDim, KnownNat hiddenDim, KnownNat batchSize)
=> GRUCell inputDim hiddenDim dtype device
-> Tensor device dtype '[batchSize, hiddenDim]	The initial Hidden state
-> [Tensor device dtype '[batchSize, inputDim]]	The list of inputs
-> Tensor device dtype '[batchSize, hiddenDim]	The final Hidden state

foldl' for lists of tensors unsing a GRUCell

gruCellScan Source #

Arguments

:: forall (inputDim :: Nat) (hiddenDim :: Nat) (batchSize :: Nat) (dtype :: DType) (device :: (DeviceType, Nat)). (KnownDType dtype, KnownNat inputDim, KnownNat hiddenDim, KnownNat batchSize)
=> GRUCell inputDim hiddenDim dtype device
-> Tensor device dtype '[batchSize, hiddenDim]	The initial Hidden state
-> [Tensor device dtype '[batchSize, inputDim]]	The list of inputs
-> [Tensor device dtype '[batchSize, hiddenDim]]	All subsequent Hidden states

scanl' for lists of tensors unsing a GRUCell