def add_activation(self: ascendie.Network, arg0: ascendie.Tensor, arg1: ascendie.ActivationKind) -> ascendie.ActivationLayer
函数功能
将激活层添加到网络中。
函数原型
add_activation(self: ascendie.Network, arg0: ascendie.Tensor, arg1: ascendie.ActivationKind) -> ascendie.ActivationLayer
约束说明
- input不可以为空，每一维不可以为0，也不可以小于-1。
- 当前不支持动态shape。
参数说明
参数名 |
参数类型 |
输入/输出 |
说明 |
|---|---|---|---|
input |
Tensor |
输入 |
激活层的输入张量。 |
kind |
ActivationKind |
输入 |
要应用的激活函数的类型。取值范围详见ascendie.ActivationKind。 |