PoolingLayer *AddPooling(Tensor *input, PoolingKind kind, Dims windowSize) noexcept;

函数功能

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

函数原型

PoolingLayer *AddPooling(Tensor *input, PoolingKind kind, Dims windowSize) noexcept;

约束说明

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

参数说明

返回值说明

返回一个PoolingLayer对象，如果添加失败时返回nullptr。