softMax function

VARP softMax(
  1. VARP logits, {
  2. int axis = -1,
})

Computes softmax activations.

Args:

  • logits: A non-empty variable. Must be Halide_Type_Float.
  • axis: The dimension softmax would be performed on. The default is -1 which indicates the last dimension.

Returns:

  • output: A variable with the same type as logits.

Implementation

VARP softMax(VARP logits, {int axis = -1}) => VARP.fromPointer(C.mnn_expr_Softmax(logits.ptr, axis));