Method: NN::ReLU#backward

Defined in:
lib/nn.rb

#backward(dout) ⇒ Object 



331
332
333
334
335
 
# File 'lib/nn.rb', line 331

def backward(dout)
  @x[@x > 0] = 1.0
  @x[@x <= 0] = 0.0
  dout * @x
end