ActivationLayer *AddActivation(Tensor *input, ActivationKind kind) noexcept;

函数功能

将激活层添加到网络中。

函数原型

ActivationLayer *AddActivation(Tensor *input, ActivationKind kind) noexcept;

约束说明

参数说明

参数名

输入/输出

说明

input

输入

激活层的输入张量。

kind

输入

要应用的激活函数的类型。取值范围详见enum class ActivationKind

返回值说明

返回新的激活层,如果添加失败时返回nullptr。