class DNN::Layers::Sqrt

Public Instance Methods

backward_node(dy) click to toggle source
# File lib/dnn/core/layers/math_layers.rb, line 189
def backward_node(dy)
  dy * (1.0 / 2 * Xumo::NMath.sqrt(@x))
end
forward_node(x) click to toggle source
# File lib/dnn/core/layers/math_layers.rb, line 184
def forward_node(x)
  @x = x
  Xumo::NMath.sqrt(x)
end