MachineLearning.NeuralNetwork.Layer

Neural Network's Layer

Documentation

data Layer

Constructors

Layer

Fields

• lUnits :: Int
• lForward :: Matrix -> Matrix -> Matrix -> Matrix
• lBackward :: Matrix -> Cache -> (Matrix, Matrix, Matrix)
• lActivation :: Matrix -> Matrix
• lActivationGradient :: Matrix -> Matrix -> Matrix
• lInitializeThetaM :: forall g. RandomGen g => (Int, Int) -> Rand g (Matrix, Matrix)

data Cache

Constructors

Cache

Fields

• cacheZ :: Matrix
• cacheX :: Matrix
• cacheW :: Matrix

affineForward :: Matrix -> Matrix -> Matrix -> Matrix

affineBackward :: Matrix R -> Cache -> (Matrix R, Matrix, Matrix R)