def add_pooling(self: Network, input: Tensor, kind: PoolingKind, windowSize: Dims) -> PoolingLayer

函数功能

在网络中添加一个多维的池化层。

函数原型

add_pooling(self: ascendie.pyascendie.Network, arg0: ascendie.pyascendie.Tensor, arg1: ascendie.pyascendie.PoolingKind, arg2: ascendie.pyascendie.Dims) -> AscendIE::PoolingLayer

约束说明

• 只支持2D池化。
• input必须是有效的Tensor对象，即必须是一个已经添加到网络中的输入Tensor，不能是空Tensor或nullptr。
• input必须是4维，DataLayout必须是NCHW。
• kind必须是PoolingKind支持的类型。
• windowSize的维度必须为2，有效取值范围为：[1, 255]，且不能大于input的H/W值。、
• 当前不支持动态shape。

参数说明