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.

条件和限制

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

TensorRT PoolingLayer相关推荐

TensorRT简介
TensorRT 介绍引用:https://arleyzhang.github.io/articles/7f4b25ce/ 1 简介 TensorRT是一个高性能的深度学习推理(Inference) ...
TensorRT优化方案图例
TensorRT优化方案图例图 12. TensorRT 循环由循环边界层设置.数据流只能通过下方式离开循环环输出层. 唯一允许的后边缘是第二个输入递归层. 图 13. 一个 if 条件构造抽象模型 ...
Tensorrt一些优化技术介绍
Tensorrt一些优化技术介绍 Figure 1. A quantizable AveragePool layer (in blue) is fused with a DQ layer and a ...
英伟达TensorRT 8-bit Inference推理
英伟达TensorRT 8-bit Inference推理引论 ● 目标:将FP32 CNN转换为INT8,不会造成显著的精度损失. ● 原因:Int8 Math具有更高的吞吐量和更低的内存需求. ...
TensorRT深度学习训练和部署图示
TensorRT深度学习训练和部署 NVIDIA TensorRT是用于生产环境的高性能深度学习推理库.功率效率和响应速度是部署的深度学习应用程序的两个关键指标,因为它们直接影响用户体验和所提供服务的 ...
TensorRT 数据和表格示例
TensorRT 数据和表格示例 TensorRT 7.1在绑定索引方面比其前身更加严格.以前,允许错误配置文件的绑定索引.考虑一个网络,该网络具有四个输入,一个输出,以及在其中的三个优化配置文件 I ...
TensorRT原理图示
TensorRT原理图示 NVIDIA的核心® TensorRT™是有助于在NVIDIA图形处理单元(GPU)的高性能推理一个C ++库.它旨在与TensorFlow,Caffe,PyTorch,MX ...
TensorRT 数据格式说明
TensorRT 数据格式说明 NVIDIA ® TensorRT™支持不同的数据格式.需要考虑两个方面:数据类型和布局. 数据类型格式数据类型是每个单独值的表示.它的大小决定了数值的范围和表示的精 ...
中继TensorRT集成
中继TensorRT集成介绍 NVIDIA TensorRT是用于优化深度学习推理的库.这种集成将使尽可能多的算子从Relay转移到TensorRT,从而无需调整调度,即可在NVIDIA GPU上提 ...

TensorRT PoolingLayer

TensorRT PoolingLayer相关推荐

最新文章

热门文章