def add_activation(self: Network, input: Tensor, kind: ActivationKind) -> ActivationLayer

函数功能

将激活层添加到网络中。

函数原型

add_activation(self: ascendie.pyascendie.Network, arg0: ascendie.pyascendie.Tensor, arg1: ascendie.pyascendie.ActivationKind) -> AscendIE::ActivationLayer

约束说明

参数说明

参数名

参数类型

输入/输出

说明

input

Tensor

输入

激活层的输入张量。

kind

ActivationKind

输入

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