def add_convolution(self: Network, input: Tensor, numOutputMap: int, kernelHW: Dims, weights: WeightsBuf, bias: WeightsBuf) -> ConvolutionLayer

函数功能

向网络添加一个卷积层。

函数原型

add_convolution(self: ascendie.pyascendie.Network, arg0: ascendie.pyascendie.Tensor, arg1: int, arg2: ascendie.pyascendie.Dims, arg3: ascendie.pyascendie.WeightsBuf, arg4: ascendie.pyascendie.WeightsBuf) -> AscendIE::ConvolutionLayer

约束说明

• 不支持动态shape或者动态batch。
• input不可以为空，数据类型必须是NCHW，维度为4维，每一维都需要大于0，且H维度取值范围在[1, 100000]，W维度取值范围在[1, 4096]。
• numOutputMap取值范围在[1, 4096]。
• kernelHW不可以为空，维度为2维，每一维都范围是[1, 255]。
• weights不可以为空，weights的value不可以为空，weights的count不可以是0。若已经设置了groupNum，weight的count需要满足公式weight.count = numOutputMap * (input -> GetDimensions() [1] / groupNum) * kernelHW[0] * kernelHW[1]。
• 如果bias不为空，bias的count需要与numOutputMap相同。

参数说明