MachineLearning.NeuralNetwork.Layer

Description

Neural Network's Layer

Documentation

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)

Constructors

Cache
Fields cacheZ :: Matrix cacheX :: Matrix cacheW :: Matrix

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