Class: NN::ReLU

Inherits:

Object

• Object
• NN::ReLU

Defined in:

lib/nn.rb

Instance Method Summary

• #backward(dout) ⇒ Object
• #forward(x) ⇒ Object

Instance Method Details

#backward(dout) ⇒ Object

# File 'lib/nn.rb', line 331

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

#forward(x) ⇒ Object

# File 'lib/nn.rb', line 325

def forward(x)
  @x = x.clone
  x[x < 0] = 0
  x
end