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

Torch.Typed.NN.Recurrent.Auxiliary

Synopsis

Documentation

data RNNInitialization Source #

Instances

Instances details
Generic RNNInitialization Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.Auxiliary

Associated Types

type Rep RNNInitialization 
Instance details

Defined in Torch.Typed.NN.Recurrent.Auxiliary

type Rep RNNInitialization = D1 ('MetaData "RNNInitialization" "Torch.Typed.NN.Recurrent.Auxiliary" "hasktorch-0.2.1.3-inplace" 'False) (C1 ('MetaCons "ConstantInitialization" 'PrefixI 'False) (U1 :: Type -> Type) :+: C1 ('MetaCons "LearnedInitialization" 'PrefixI 'False) (U1 :: Type -> Type))
Show RNNInitialization Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.Auxiliary

type Rep RNNInitialization Source # 
Instance details

Defined in Torch.Typed.NN.Recurrent.Auxiliary

type Rep RNNInitialization = D1 ('MetaData "RNNInitialization" "Torch.Typed.NN.Recurrent.Auxiliary" "hasktorch-0.2.1.3-inplace" 'False) (C1 ('MetaCons "ConstantInitialization" 'PrefixI 'False) (U1 :: Type -> Type) :+: C1 ('MetaCons "LearnedInitialization" 'PrefixI 'False) (U1 :: Type -> Type))

calculateFan :: [Int] -> (Int, Int) Source #

Fan-in / Fan-out scaling calculation

xavierUniformFIXME :: Tensor -> Float -> [Int] -> IO Tensor Source #

Xavier Initialization - Uniform