Module: DNN::Layers::MergeLayerNode

Instance Method Summary collapse

Instance Method Details

#backward(dy) ⇒ Object



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

def backward(dy)
  backward_node(dy)
end

#backward_node(dy) ⇒ Object

Raises:

  • (NotImplementedError)


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

def backward_node(dy)
  raise NotImplementedError, "Class '#{self.class.name}' has implement method 'backward_node'"
end

#forward(input1, input2) ⇒ Object



5
6
7
8
9
10
11
12
13
# File 'lib/dnn/core/layers/merge_layers.rb', line 5

def forward(input1, input2)
  x1 = input1.data
  x2 = input2.data
  prev1 = (input1.is_a?(Tensor) ? input1.link : input1)
  prev2 = (input2.is_a?(Tensor) ? input2.link : input2)
  y = forward_node(x1, x2)
  link = TwoInputLink.new(prev1, prev2, self)
  Tensor.new(y, link)
end

#forward_node(x1, x2) ⇒ Object

Raises:

  • (NotImplementedError)


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

def forward_node(x1, x2)
  raise NotImplementedError, "Class '#{self.class.name}' has implement method 'forward_node'"
end