Torch.Typed.NN.Recurrent.Auxiliary

Documentation

data RNNInitialization

Constructors

ConstantInitialization

LearnedInitialization

Instances

Generic RNNInitialization

Defined in Torch.Typed.NN.Recurrent.Auxiliary

Associated Types

type Rep RNNInitialization

Defined in Torch.Typed.NN.Recurrent.Auxiliary

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

Methods

from :: RNNInitialization -> Rep RNNInitialization x

to :: Rep RNNInitialization x -> RNNInitialization

Show RNNInitialization

Defined in Torch.Typed.NN.Recurrent.Auxiliary

Methods

showsPrec :: Int -> RNNInitialization -> ShowS

show :: RNNInitialization -> String

showList :: [RNNInitialization] -> ShowS

type Rep RNNInitialization

Defined in Torch.Typed.NN.Recurrent.Auxiliary

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

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

Fan-in / Fan-out scaling calculation

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

Xavier Initialization - Uniform