枚举类型,激活函数选择。
enum class ActivationKind : int32_t { RELU = 0, SIGMOID, LEAKY_RELU, HARD_SIGMOID, TANH, SWISH, };
成员名称 |
描述 |
---|---|
RELU |
表示ReLU激活函数。 |
SIGMOID |
表示Sigmoid激活函数。 |
LEAKY_RELU |
表示LeakyRelu激活函数。 |
HARD_SIGMOID |
表示Hard Sigmoid激活函数。 |
TANH |
表示Tanh激活函数。 |
SWISH |
表示Swish激活函数。 |