Class: GRX::NN::LeakyReLU

Inherits:

Module

• Object
• Module
• GRX::NN::LeakyReLU

Defined in:

lib/grx/nn.rb

Instance Method Summary

• #forward(x) ⇒ Object
• #initialize(alpha = 0.01) ⇒ LeakyReLU constructor

  A new instance of LeakyReLU.

• #to_s ⇒ Object

Methods inherited from Module

#call, #parameters, #zero_grad

Constructor Details

#initialize(alpha = 0.01) ⇒ LeakyReLU

Returns a new instance of LeakyReLU.

# File 'lib/grx/nn.rb', line 144

def initialize(alpha = 0.01)
  @alpha = alpha
end

Instance Method Details

#forward(x) ⇒ Object

# File 'lib/grx/nn.rb', line 147

def forward(x) = x.leaky_relu(@alpha)

#to_s ⇒ Object

# File 'lib/grx/nn.rb', line 148

def to_s = "LeakyReLU(alpha=#{@alpha})"