Function _Selu

express/NeuralNetWorkOp.cpp:986–996

Computes scaled exponential linear: scale * alpha * (exp(features) - 1) if < 0, scale * features otherwise. Args: features: A variable of type Halide_Type_Float scale: Scaling factor (positive float) alpha: Alpha factor (positive float) Returns: A variable. Has the same type as features. */

A variable. Has the same type as features.
*/
VARP _Selu(VARP features, float scale, float alpha) {
    std::unique_ptr<OpT> op(new OpT);
    op->type       = OpType_Selu;
    auto seluParam = new SeluT;
    op->main.type = OpParameter_Selu;
    seluParam->scale = scale;
    seluParam->alpha = alpha;
    op->main.value = seluParam;
    return (Variable::create(Expr::create(std::move(op), {features})));

}
/*Gather slices from params into a variable with shape specified by indices.
Args:
params: A variable. The variables from which to gather values.