def add_pooling(self: ascendie.Network, input: ascendie.Tensor, kind: ascendie.PoolingKind, windowSize: ascendie.Dims, ascendie.Dims = ascendie.Dims([])) -> ascendie.PoolingLayer

函数功能

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

函数原型

add_pooling(self: ascendie.Network, input: ascendie.Tensor, kind: ascendie.PoolingKind, windowSize: ascendie.Dims, ascendie.Dims = ascendie.Dims([])) -> ascendie.PoolingLayer

约束说明

• ADAPTIVE_AVERAGE场景只支持2D输入，其他场景支持2D和3D输入。
• input必须是有效的Tensor对象，即必须是一个已经添加到网络中的输入Tensor，不能是空Tensor或nullptr。
• input的DataLayout必须是NCHW。
• kind必须是PoolingKind支持的类型。
• windowSize的维度中，ADAPTIVE_AVERAGE类型输入{}，MAX和AVERAGE类型设置维度为2/3维，有效取值范围为：[1, 255]，且不能大于input的H/W值。
• 当前不支持动态shape。

参数说明