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