Class: DNN::Link

Inherits:

Object

• Object
• DNN::Link

Defined in:

lib/dnn/core/link.rb

Instance Attribute Summary

• #layer ⇒ Object

  Returns the value of attribute layer.

• #prev ⇒ Object

  Returns the value of attribute prev.

Instance Method Summary

• #backward(dy) ⇒ Object
• #initialize(prev = nil, layer = nil) ⇒ Link constructor

  A new instance of Link.

Constructor Details

#initialize(prev = nil, layer = nil) ⇒ Link

Returns a new instance of Link.

# File 'lib/dnn/core/link.rb', line 6

def initialize(prev = nil, layer = nil)
  @prev = prev
  @layer = layer
end

Instance Attribute Details

#layer ⇒ Object

Returns the value of attribute layer.

# File 'lib/dnn/core/link.rb', line 4

def layer
  @layer
end

#prev ⇒ Object

Returns the value of attribute prev.

# File 'lib/dnn/core/link.rb', line 3

def prev
  @prev
end

Instance Method Details

#backward(dy) ⇒ Object

# File 'lib/dnn/core/link.rb', line 11

def backward(dy)
  dy = @layer.backward(dy)
  @prev&.backward(dy)
end