| Safe Haskell | Safe-Inferred |
|---|---|
| Language | Haskell2010 |
Torch.Initializers
Synopsis
- data NonLinearity
- data FanMode
- newtype Shape = Shape [Int]
- calculateGain :: NonLinearity -> Float
- calculateFan :: [Int] -> (Int, Int)
- xavierUniform :: Float -> [Int] -> IO Tensor
- xavierNormal :: Float -> [Int] -> IO Tensor
- getter :: FanMode -> (Int, Int) -> Int
- kaimingUniform :: FanMode -> NonLinearity -> [Int] -> IO Tensor
- kaimingNormal :: FanMode -> NonLinearity -> [Int] -> IO Tensor
- kaimingFC :: [Int] -> IO (Tensor, Tensor)
- kaimingUniform' :: [Int] -> IO Tensor
- kaimingNormal' :: [Int] -> IO Tensor
- xavierUniform' :: [Int] -> IO Tensor
- xavierNormal' :: [Int] -> IO Tensor
Documentation
calculateGain :: NonLinearity -> Float Source #
Gain scaling value for He initialization
getter :: FanMode -> (Int, Int) -> Int Source #
Get fan in or fan out value depending on selected fan mode, used by Kaiming
kaimingUniform :: FanMode -> NonLinearity -> [Int] -> IO Tensor Source #
Kaiming Initialization - Uniform
kaimingNormal :: FanMode -> NonLinearity -> [Int] -> IO Tensor Source #
Kaiming Initialization - Normal