Function _Softmax

express/NeuralNetWorkOp.cpp:479–486

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`. */

output: A variable with the same type as `logits`.
*/
VARP _Softmax(VARP logits, int axis) {
    std::unique_ptr<OpT> softmax(new OpT);
    softmax->type                = OpType_Softmax;
    softmax->main.type           = OpParameter_Axis;
    softmax->main.value          = new AxisT;
    softmax->main.AsAxis()->axis = axis;
    return (Variable::create(Expr::create(softmax.get(), {logits})));
}
/*Computes softplus: log(exp(features) + 1).
Args:
features: A variable. Must be Halide_Type_Float.