class NN::ReLU

Public Instance Methods

backward(dout) click to toggle source
# File lib/nn.rb, line 331
def backward(dout)
  @x[@x > 0] = 1.0
  @x[@x <= 0] = 0.0
  dout * @x
end
forward(x) click to toggle source
# File lib/nn.rb, line 325
def forward(x)
  @x = x.clone
  x[x < 0] = 0
  x
end