ActivationLayer *AddActivation(Tensor *input, ActivationKind kind) noexcept;
函数功能
将激活层添加到网络中。
函数原型
ActivationLayer *AddActivation(Tensor *input, ActivationKind kind) noexcept;
约束说明
- input维度必须在[1, 8]。
- 当前不支持动态shape。
参数说明
参数名 |
输入/输出 |
说明 |
|---|---|---|
input |
输入 |
激活层的输入张量。 |
kind |
输入 |
要应用的激活函数的类型。取值范围详见enum class ActivationKind。 |
返回值说明
返回新的激活层，如果添加失败则返回nullptr或抛出异常。
父主题： class Network