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或抛出异常。
父主题: class Network