在网络中添加一个多维的池化层。
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。