Method: NN::Softmax#forward

Defined in:
lib/nn.rb

#forward(x) ⇒ Object 



359
360
361
 
# File 'lib/nn.rb', line 359

def forward(x)
  @out = NMath.exp(x) / NMath.exp(x).sum(1).reshape(x.shape[0], 1)
end