下载
中文
注册

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

函数功能

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

函数原型

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

约束说明

  • 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值。

参数说明

参数名

输入/输出

说明

input

输入

池化层的输入张量。

kind

输入

池化类型,详见enum class PoolingKind

windowSize

输入

池化窗口大小。

outputSize

输入

输出特征图大小。默认为Dims(),池化类型为ADAPTIVE_AVERAGE时需传入具体的outputSize。

返回值说明

返回一个PoolingLayer对象,如果添加失败则返回nullptr或抛出异常。