下载
中文
注册

enum class ActivationKind

说明

枚举类型,激活函数选择。

定义

enum class ActivationKind : int32_t {
    RELU = 0,
    SIGMOID,
    LEAKY_RELU,
    HARD_SIGMOID,
    TANH,
    SWISH,
    GELU,
    ELU,
    SELU,
    SOFTPLUS, 
    MISH,
    RELU6
};

成员

成员名称

描述

RELU

表示ReLU激活函数。

SIGMOID

表示Sigmoid激活函数。

LEAKY_RELU

表示LeakyRelu激活函数。

HARD_SIGMOID

表示Hard Sigmoid激活函数。

TANH

表示Tanh激活函数。

SWISH

表示Swish激活函数。

GELU

表示Gelu激活函数。

ELU

表示Elu激活函数。

SELU

表示Selu激活函数。

SOFTPLUS

表示Softplus激活函数。

MISH

表示Mish激活函数。

RELU6

表示Relu6激活函数。