Class: DNN::Layers::Layer

Inherits:

Object

• Object
• DNN::Layers::Layer

Includes:

Numo

Defined in:

lib/dnn/core/layers.rb

Overview

Super class of all optimizer classes.

Direct Known Subclasses

Dropout, Flatten, HasParamLayer, InputLayer, MaxPool2D, OutputLayer, Reshape

Instance Method Summary

• #backward ⇒ Object

  Backward propagation.

• #forward ⇒ Object

  Forward propagation.

• #init(model) ⇒ Object

  Initialize layer when model is compiled.

• #prev_layer ⇒ Object

  Get the previous layer.

• #shape ⇒ Object

  Get the shape of the layer.

Instance Method Details

#backward ⇒ Object

Backward propagation.

# File 'lib/dnn/core/layers.rb', line 17

def backward() end

#forward ⇒ Object

Forward propagation.

# File 'lib/dnn/core/layers.rb', line 14

def forward() end

#init(model) ⇒ Object

Initialize layer when model is compiled.

# File 'lib/dnn/core/layers.rb', line 9

def init(model)
  @model = model
end

#prev_layer ⇒ Object

Get the previous layer.

# File 'lib/dnn/core/layers.rb', line 25

def prev_layer
  @model.layers[@model.layers.index(self) - 1]
end

#shape ⇒ Object

Get the shape of the layer.

# File 'lib/dnn/core/layers.rb', line 20

def shape
  prev_layer.shape
end