Class: GRX::NN::Softmax

Inherits:

Module

• Object
• Module
• GRX::NN::Softmax

Defined in:

lib/grx/nn.rb

Instance Method Summary

• #forward(x) ⇒ Object
• #to_s ⇒ Object

Methods inherited from Module

#call, #parameters, #zero_grad

Instance Method Details

#forward(x) ⇒ Object

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

def forward(x) = x.softmax

#to_s ⇒ Object

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

def to_s = "Softmax()"