将激活层添加到网络中。
ActivationLayer *AddActivation(Tensor *input, ActivationKind kind) noexcept;
参数名 |
输入/输出 |
说明 |
---|---|---|
input |
输入 |
激活层的输入张量。 |
kind |
输入 |
要应用的激活函数的类型。取值范围详见enum class ActivationKind。 |
返回新的激活层,如果添加失败时返回nullptr。