Module: DNN::Layers::LayerNode

Included in:
AvgPool2D, BatchNormalization, Conv2D, Conv2DTranspose, Dense, Dropout, ELU, Exp, Flatten, GRUDense, InputLayer, LSTMDense, Lasso, LeakyReLU, Log, MaxPool2D, Mean, Mish, Pool2D, Pow, RNN, ReLU, Reshape, Ridge, Sigmoid, SimpleRNNDense, Softplus, Softsign, Sqrt, Sum, Swish, Tanh, UnPool2D
Defined in:
lib/dnn/core/layers/basic_layers.rb

Instance Method Summary collapse

Instance Method Details

#backward(dy) ⇒ Object 



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

def backward(dy)
  backward_node(dy)
end

#backward_node(dy) ⇒ Object

Raises:

  • (NotImplementedError) 


21
22
23
 
# File 'lib/dnn/core/layers/basic_layers.rb', line 21

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

#forward(input_tensor) ⇒ Object 



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

def forward(input_tensor)
  x = input_tensor.data
  prev_link = (input_tensor.is_a?(Tensor) ? input_tensor.link : input_tensor)
  y = forward_node(x)
  link = Link.new(prev_link, self)
  Tensor.new(y, link)
end

#forward_node(x) ⇒ Object

Raises:

  • (NotImplementedError) 


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

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