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

函数功能

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

函数原型

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

约束说明

参数说明

参数名

输入/输出

说明

input

输入

池化层的输入张量。

kind

输入

池化类型,详见enum class PoolingKind

windowSize

输入

池化窗口大小。

outputSize

输入

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

返回值说明

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