Method: NN::ReLU#backward

Defined in:
lib/nn.rb

#backward(dout) ⇒ Object 



327
328
329
330
331
 
# File 'lib/nn.rb', line 327

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