arsenal.nn.activation

Module Contents

arsenal.nn.activation.softmax(xs, axis=- 1)[source]

Apply softmax function over a dimension of xs.

Parameters

  • xs – Array of floats to softmax.

  • axis – Dimension to softmax

Returns

Softmaxed xs