网络定义中的池化层。更多...

#include <NvInfer.h>

nvinfer1::IPoolingLayer 的继承关系图

公共成员函数
void	setPoolingType (PoolingType type) noexcept
	设置要执行的激活类型。更多...

PoolingType	getPoolingType () const noexcept
	获取要执行的激活类型。更多...

void	setBlendFactor (float blendFactor) noexcept
	设置 max_average_blend 模式的混合因子：max_average_blendPool = (1-blendFactor)maxPool + blendFactoravgPool。blendFactor 是 [0,1] 范围内的用户值，默认值为 0.0。此值仅适用于 kMAX_AVERAGE_BLEND 模式。更多...

float	getBlendFactor () const noexcept
	获取 max_average_blend 模式的混合因子：max_average_blendPool = (1-blendFactor)maxPool + blendFactoravgPool。blendFactor 是 [0,1] 范围内的用户值，默认值为 0.0。在 kMAX_AVERAGE_BLEND 以外的模式中，blendFactor 将被忽略。更多...

void	setAverageCountExcludesPadding (bool exclusive) noexcept
	设置平均池化是否使用窗口和未填充输入之间的重叠区域作为分母。如果未设置，则分母是池化窗口和填充输入之间的重叠区域。更多...

bool	getAverageCountExcludesPadding () const noexcept
	获取平均池化是否使用窗口和未填充输入之间的重叠区域作为分母。更多...

void	setPrePadding (Dims const &padding) noexcept
	设置池化的多维前填充。更多...

Dims	getPrePadding () const noexcept
	获取前填充。更多...

void	setPostPadding (Dims const &padding) noexcept
	设置池化的多维后填充。更多...

Dims	getPostPadding () const noexcept
	获取后填充。更多...

void	setPaddingMode (PaddingMode paddingMode) noexcept
	设置填充模式。更多...

PaddingMode	getPaddingMode () const noexcept
	获取填充模式。更多...

void	setWindowSizeNd (Dims const &windowSize) noexcept
	设置池化的多维窗口大小。更多...

Dims	getWindowSizeNd () const noexcept
	获取池化的多维窗口大小。更多...

void	setStrideNd (Dims const &stride) noexcept
	设置池化的多维步长。更多...

Dims	getStrideNd () const noexcept
	获取池化的多维步长。更多...

void	setPaddingNd (Dims const &padding) noexcept
	设置池化的多维填充。更多...

Dims	getPaddingNd () const noexcept
	获取池化的多维填充。更多...

继承自 nvinfer1::ILayer 的公共成员函数
LayerType	getType () const noexcept
	返回层的类型。更多...

void	setName (char const *name) noexcept
	设置层的名称。更多...

char const *	getName () const noexcept
	返回层的名称。更多...

int32_t	getNbInputs () const noexcept
	获取层的输入数量。更多...

ITensor *	getInput (int32_t index) const noexcept
	获取与给定索引对应的层输入。更多...

int32_t	getNbOutputs () const noexcept
	获取层的输出数量。更多...

ITensor *	getOutput (int32_t index) const noexcept
	获取与给定索引对应的层输出。更多...

void	setInput (int32_t index, ITensor &tensor) noexcept
	用特定张量替换此层的输入。更多...

void	setPrecision (DataType dataType) noexcept
	在弱类型网络中设置此层的首选或必需的计算精度。更多...

DataType	getPrecision () const noexcept
	获取此层的计算精度更多...

bool	precisionIsSet () const noexcept
	是否已为此层设置计算精度更多...

void	resetPrecision () noexcept
	重置此层的计算精度更多...

void	setOutputType (int32_t index, DataType dataType) noexcept
	在弱类型网络中设置此层的输出类型。更多...

DataType	getOutputType (int32_t index) const noexcept
	获取此层的输出类型更多...

bool	outputTypeIsSet (int32_t index) const noexcept
	是否已为此层设置输出类型更多...

void	resetOutputType (int32_t index) noexcept
	重置此层的输出类型更多...

void	setMetadata (char const *metadata) noexcept
	为此层设置元数据。更多...

char const *	getMetadata () const noexcept
	获取层的元数据。更多...

保护成员函数
virtual	~IPoolingLayer () noexcept=default

继承自 nvinfer1::ILayer 的保护成员函数
virtual	~ILayer () noexcept=default

继承自 nvinfer1::INoCopy 的保护成员函数
	INoCopy ()=default

virtual	~INoCopy ()=default

	INoCopy (INoCopy const &other)=delete

INoCopy &	operator= (INoCopy const &other)=delete

	INoCopy (INoCopy &&other)=delete

INoCopy &	operator= (INoCopy &&other)=delete

保护属性
apiv::VPoolingLayer *	mImpl

继承自 nvinfer1::ILayer 的保护属性
apiv::VLayer *	mLayer

详细描述

网络定义中的池化层。

该层在输入上的窗口内应用缩减操作。

警告: 在 Int8 模式下使用 DeviceType::kDLA 运行池化层时，输入和输出张量的动态范围必须相等。; 请勿从此类继承，这样做会破坏 API 和 ABI 的向前兼容性。

构造函数 & 析构函数文档

◆ ~IPoolingLayer()

virtual nvinfer1::IPoolingLayer::~IPoolingLayer ( )

protectedvirtualdefaultnoexcept

成员函数文档

◆ getAverageCountExcludesPadding()

bool nvinfer1::IPoolingLayer::getAverageCountExcludesPadding ( ) const

inlinenoexcept

获取平均池化是否使用窗口和未填充输入之间的重叠区域作为分母。

另请参阅: setAverageCountExcludesPadding()

◆ getBlendFactor()

float nvinfer1::IPoolingLayer::getBlendFactor ( ) const

inlinenoexcept

获取 max_average_blend 模式的混合因子：max_average_blendPool = (1-blendFactor)*maxPool + blendFactor*avgPool。blendFactor 是 [0,1] 范围内的用户值，默认值为 0.0。在 kMAX_AVERAGE_BLEND 以外的模式中，blendFactor 将被忽略。

另请参阅: setBlendFactor()

◆ getPaddingMode()

PaddingMode nvinfer1::IPoolingLayer::getPaddingMode ( ) const

inlinenoexcept

获取填充模式。

默认值：kEXPLICIT_ROUND_DOWN

另请参阅: setPaddingMode()

◆ getPaddingNd()

Dims nvinfer1::IPoolingLayer::getPaddingNd ( ) const

inlinenoexcept

获取池化的多维填充。

如果填充是不对称的，则返回前填充。

另请参阅: setPaddingNd()

◆ getPoolingType()

PoolingType nvinfer1::IPoolingLayer::getPoolingType ( ) const

inlinenoexcept

获取要执行的激活类型。

另请参阅: setPoolingType(), PoolingType

◆ getPostPadding()

Dims nvinfer1::IPoolingLayer::getPostPadding ( ) const

inlinenoexcept

获取后填充。

另请参阅: setPostPadding()

◆ getPrePadding()

Dims nvinfer1::IPoolingLayer::getPrePadding ( ) const

inlinenoexcept

获取前填充。

另请参阅: setPrePadding()

◆ getStrideNd()

Dims nvinfer1::IPoolingLayer::getStrideNd ( ) const

inlinenoexcept

获取池化的多维步长。

另请参阅: setStrideNd()

◆ getWindowSizeNd()

Dims nvinfer1::IPoolingLayer::getWindowSizeNd ( ) const

inlinenoexcept

获取池化的多维窗口大小。

另请参阅: setWindowSizeNd()

◆ setAverageCountExcludesPadding()

void nvinfer1::IPoolingLayer::setAverageCountExcludesPadding ( bool exclusive )

inlinenoexcept

设置平均池化是否使用窗口和未填充输入之间的重叠区域作为分母。如果未设置，则分母是池化窗口和填充输入之间的重叠区域。

默认值：true

另请参阅: getAverageCountExcludesPadding()

◆ setBlendFactor()

void nvinfer1::IPoolingLayer::setBlendFactor ( float blendFactor )

inlinenoexcept

设置 max_average_blend 模式的混合因子：max_average_blendPool = (1-blendFactor)*maxPool + blendFactor*avgPool。blendFactor 是 [0,1] 范围内的用户值，默认值为 0.0。此值仅适用于 kMAX_AVERAGE_BLEND 模式。

由于 DLA 不支持 kMAX_AVERAGE_BLEND，因此在 DLA 上 blendFactor 将被忽略。

另请参阅: getBlendFactor()

◆ setPaddingMode()

void nvinfer1::IPoolingLayer::setPaddingMode ( PaddingMode paddingMode )

inlinenoexcept

设置填充模式。

如果同时使用 setPaddingMode 和 setPre/PostPadding，则填充模式优先。

默认值：kEXPLICIT_ROUND_DOWN

另请参阅: getPaddingMode()

◆ setPaddingNd()

void nvinfer1::IPoolingLayer::setPaddingNd ( Dims const & padding )

inlinenoexcept

设置池化的多维填充。

输入将在每个维度上填充此数量的元素。填充是对称的。填充值取决于池化类型，最大池化使用 -inf，平均池化使用零填充。

默认值：(0, 0, ..., 0)

如果在 DLA 上执行此层，则仅支持 2D 填充，填充的高度和宽度都必须在 [0,7] 范围内。

另请参阅: getPaddingNd() setPadding() getPadding()

◆ setPoolingType()

void nvinfer1::IPoolingLayer::setPoolingType ( PoolingType type )

inlinenoexcept

设置要执行的激活类型。

DLA 仅支持 kMAX 和 kAVERAGE 池化类型。

另请参阅: getPoolingType(), PoolingType

◆ setPostPadding()

void nvinfer1::IPoolingLayer::setPostPadding ( Dims const & padding )

inlinenoexcept

设置池化的多维后填充。

输入的末尾将在每个维度上填充此数量的元素。填充值取决于池化类型，最大池化使用 -inf，平均池化使用零填充。

默认值：(0, 0, ..., 0)

如果在 DLA 上执行此层，则仅支持 2D 填充，填充的高度和宽度都必须在 [0,7] 范围内。

另请参阅: getPostPadding()

◆ setPrePadding()

void nvinfer1::IPoolingLayer::setPrePadding ( Dims const & padding )

inlinenoexcept

设置池化的多维前填充。

输入的开始将在每个维度上填充此数量的元素。填充值取决于池化类型，最大池化使用 -inf，平均池化使用零填充。

默认值：(0, 0, ..., 0)

如果在 DLA 上执行此层，则仅支持 2D 填充，填充的高度和宽度都必须在 [0,7] 范围内。

另请参阅: getPrePadding()

◆ setStrideNd()

void nvinfer1::IPoolingLayer::setStrideNd ( Dims const & stride )

inlinenoexcept

设置池化的多维步长。

默认值：(1, 1, ..., 1)

如果在 DLA 上执行此层，则仅支持 2D 步长，步长的高度和宽度都必须在 [1,16] 范围内。

另请参阅: getStrideNd()

◆ setWindowSizeNd()

void nvinfer1::IPoolingLayer::setWindowSizeNd ( Dims const & windowSize )

inlinenoexcept

设置池化的多维窗口大小。

如果在 DLA 上执行此层，则仅支持 2D 窗口大小，窗口大小的高度和宽度都必须在 [1,8] 范围内。

另请参阅: getWindowSizeNd() setWindowSize() getWindowSize()

成员数据文档

◆ mImpl

apiv::VPoolingLayer* nvinfer1::IPoolingLayer::mImpl

protected

此类文档的生成文件为：

NvInfer.h

公共成员函数

保护成员函数

保护属性

详细描述

构造函数 & 析构函数文档

◆ ~IPoolingLayer()

成员函数文档

◆ getAverageCountExcludesPadding()

◆ getBlendFactor()

◆ getPaddingMode()

◆ getPaddingNd()

◆ getPoolingType()

◆ getPostPadding()

◆ getPrePadding()

◆ getStrideNd()

◆ getWindowSizeNd()

◆ setAverageCountExcludesPadding()

◆ setBlendFactor()

◆ setPaddingMode()

◆ setPaddingNd()

◆ setPoolingType()

◆ setPostPadding()

◆ setPrePadding()

◆ setStrideNd()

◆ setWindowSizeNd()

成员数据文档

◆ mImpl