activation

Documentation

Index

• type AvgPooling
  • func (e AvgPooling) Backward(error [][][]float64, lr float64) [][][]float64
  • func (e AvgPooling) Forward(in [][][]float64) [][][]float64
• type MaxPooling
  • func (e *MaxPooling) Backward(error [][][]float64, lr float64) [][][]float64
  • func (e *MaxPooling) Forward(in [][][]float64) [][][]float64
• type ReLu
  • func (e *ReLu) Backward(err [][][]float64, lr float64) [][][]float64
  • func (e *ReLu) Forward(in [][][]float64) [][][]float64
• type Sigmoid
• type Tanh
  • func (e Tanh) Backward(err [][][]float64, lr float64) [][][]float64
  • func (e *Tanh) Forward(in [][][]float64) [][][]float64

Types

type AvgPooling

type AvgPooling struct {
	Ksize  int
	Stride int
}

Average pooling layer

func (AvgPooling) Backward

func (e AvgPooling) Backward(error [][][]float64, lr float64) [][][]float64

func (AvgPooling) Forward

func (e AvgPooling) Forward(in [][][]float64) [][][]float64

Average pooling forward function

type MaxPooling

type MaxPooling struct {
	// contains filtered or unexported fields
}

Maximum pooling layer

func (*MaxPooling) Backward

func (e *MaxPooling) Backward(error [][][]float64, lr float64) [][][]float64

func (*MaxPooling) Forward

func (e *MaxPooling) Forward(in [][][]float64) [][][]float64

type ReLu

type ReLu struct {
	// contains filtered or unexported fields
}

Rectifier linear unit layer

func (*ReLu) Backward

func (e *ReLu) Backward(err [][][]float64, lr float64) [][][]float64

ReLu backward function

func (*ReLu) Forward

func (e *ReLu) Forward(in [][][]float64) [][][]float64

ReLu forward

type Sigmoid

type Sigmoid struct {
	Input [][][]float64
}

type Tanh

type Tanh struct {
	Input [][][]float64
}

Hyperbolic tangent layer

func (Tanh) Backward

func (e Tanh) Backward(err [][][]float64, lr float64) [][][]float64

Tanh backward function

func (*Tanh) Forward

func (e *Tanh) Forward(in [][][]float64) [][][]float64

Tanh Forward function