grenade-0.1.0: Practical Deep Learning in Haskell

Safe HaskellNone
LanguageHaskell98

Grenade.Layers.FullyConnected

Synopsis

Documentation

data FullyConnected i o Source #

A basic fully connected (or inner product) neural network layer.

Constructors

FullyConnected !(FullyConnected' i o) !(FullyConnected' i o) 

Instances

Show (FullyConnected i o) Source # 
(KnownNat i, KnownNat o) => Serialize (FullyConnected i o) Source # 

Methods

put :: Putter (FullyConnected i o) #

get :: Get (FullyConnected i o) #

(KnownNat i, KnownNat o) => UpdateLayer (FullyConnected i o) Source # 
(KnownNat i, KnownNat o) => Layer (FullyConnected i o) (D1 i) (D1 o) Source # 

Associated Types

type Tape (FullyConnected i o) (D1 i :: Shape) (D1 o :: Shape) :: * Source #

Methods

runForwards :: FullyConnected i o -> S (D1 i) -> (Tape (FullyConnected i o) (D1 i) (D1 o), S (D1 o)) Source #

runBackwards :: FullyConnected i o -> Tape (FullyConnected i o) (D1 i) (D1 o) -> S (D1 o) -> (Gradient (FullyConnected i o), S (D1 i)) Source #

type Gradient (FullyConnected i o) Source # 
type Tape (FullyConnected i o) (D1 i) (D1 o) Source # 
type Tape (FullyConnected i o) (D1 i) (D1 o) = S (D1 i)

data FullyConnected' i o Source #

Constructors

FullyConnected' !(R o) !(L o i)