TensorRT PoolingLayer

TensorRT PoolingLayer

IPoolingLayer在通道中实现池。支持的池类型有maximum、average和maximum average混合。

Layer Description: 2D pooling层描述:二维池

用2D滤波器计算a维张量a上的池,生成B维的张量B。B的维数取决于a的维数、窗口大小r、对称填充p和步长s,因此:
在这里插入图片描述
PoolingType::kMAX

Maximum over elements in window.

PoolingType::kAVERAGE

Average over elements in the window.

PoolingType::kMAX_AVERAGE_BLEND

Hybrid of maximum and average pooling. The
results of the maximum pooling and the average pooling are combined with the blending factor as (1-blendFactor)*maximumPoolingResult

  • blendFactor*averagePoolingResult to yield the result. The blendFactor
    can be set to a value between 0 and 1.

默认情况下,平均池是在池窗口和填充的输入之间的重叠处执行的。如果exclusive参数设置为true,则在池窗口和未添加的输入之间的重叠区域执行平均池。

Layer Description: 3D pooling图层说明:三维池

使用3D过滤器计算a维度张量a上的池,以生成维度B的张量B。B的维度取决于a的维度、窗口大小r、对称填充p和跨距s,因此:
在这里插入图片描述

Where func is defined by one of the pooling types t:

PoolingType::kMAX

Maximum over elements in window.

PoolingType::kAVERAGE

Average over elements in the window.

PoolingType::kMAX_AVERAGE_BLEND

Hybrid of maximum and average pooling. The results of the maximum pooling and the average pooling are combined with the blending factor as (1-blendFactor)*maximumPoolingResult

  • blendFactor*averagePoolingResult to yield the result. The blendFactor can be set to a value between 0 and 1.

默认情况下,平均池是在池窗口和填充的输入之间的重叠处执行的。如果exclusive参数设置为true,则在池窗口和未添加的输入之间的重叠区域执行平均池。

条件和限制

2D或3D由输入核维数的数量决定。对于2D池,输入和输出张量应该有3个或更多维。对于3D池,输入和输出张量应具有4个或更多维度。

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 深蓝海洋 设计师:CSDN官方博客 返回首页