网络定义中的反卷积层。更多...

#include <NvInfer.h>

nvinfer1::IDeconvolutionLayer 的继承关系图

公共成员函数
void	setNbOutputMaps (int64_t nbOutputMaps) noexcept
	设置反卷积的输出特征图数量。更多...

int64_t	getNbOutputMaps () const noexcept
	获取反卷积的输出特征图数量。更多...

void	setNbGroups (int64_t nbGroups) noexcept
	设置反卷积的分组数。更多...

int64_t	getNbGroups () const noexcept
	获取反卷积的分组数。更多...

void	setKernelWeights (Weights weights) noexcept
	设置反卷积的内核权重。更多...

Weights	getKernelWeights () const noexcept
	获取反卷积的内核权重。更多...

void	setBiasWeights (Weights weights) noexcept
	设置反卷积的偏置权重。更多...

Weights	getBiasWeights () 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	setKernelSizeNd (Dims const &kernelSize) noexcept
	设置反卷积的多维内核大小。更多...

Dims	getKernelSizeNd () const noexcept
	获取反卷积的多维内核大小。更多...

void	setStrideNd (Dims const &stride) noexcept
	设置反卷积的多维步长。更多...

Dims	getStrideNd () const noexcept
	获取反卷积的多维步长。更多...

void	setPaddingNd (Dims const &padding) noexcept
	设置反卷积的多维填充。更多...

Dims	getPaddingNd () const noexcept
	获取反卷积的多维填充。更多...

void	setDilationNd (Dims const &dilation) noexcept
	设置反卷积的多维膨胀率。更多...

Dims	getDilationNd () const noexcept
	获取反卷积的多维膨胀率。更多...

void	setInput (int32_t index, ITensor &tensor) 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	~IDeconvolutionLayer () 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::VDeconvolutionLayer *	mImpl

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

详细描述

网络定义中的反卷积层。

警告: 请勿从此类继承，否则会破坏 API 和 ABI 的向前兼容性。

构造函数 & 析构函数文档

◆ ~IDeconvolutionLayer()

virtual nvinfer1::IDeconvolutionLayer::~IDeconvolutionLayer ( )

protectedvirtualdefaultnoexcept

成员函数文档

◆ getBiasWeights()

Weights nvinfer1::IDeconvolutionLayer::getBiasWeights ( ) const

inlinenoexcept

获取反卷积的偏置权重。

另请参阅: getBiasWeights()

◆ getDilationNd()

Dims nvinfer1::IDeconvolutionLayer::getDilationNd ( ) const

inlinenoexcept

获取反卷积的多维膨胀率。

另请参阅: setDilationNd()

◆ getKernelSizeNd()

Dims nvinfer1::IDeconvolutionLayer::getKernelSizeNd ( ) const

inlinenoexcept

获取反卷积的多维内核大小。

另请参阅: setKernelSizeNd()

◆ getKernelWeights()

Weights nvinfer1::IDeconvolutionLayer::getKernelWeights ( ) const

inlinenoexcept

获取反卷积的内核权重。

另请参阅: setNbGroups()

◆ getNbGroups()

int64_t nvinfer1::IDeconvolutionLayer::getNbGroups ( ) const

inlinenoexcept

获取反卷积的分组数。

另请参阅: setNbGroups()

◆ getNbOutputMaps()

int64_t nvinfer1::IDeconvolutionLayer::getNbOutputMaps ( ) const

inlinenoexcept

获取反卷积的输出特征图数量。

另请参阅: setNbOutputMaps()

◆ getPaddingMode()

PaddingMode nvinfer1::IDeconvolutionLayer::getPaddingMode ( ) const

inlinenoexcept

获取填充模式。

默认值：kEXPLICIT_ROUND_DOWN

另请参阅: setPaddingMode()

◆ getPaddingNd()

Dims nvinfer1::IDeconvolutionLayer::getPaddingNd ( ) const

inlinenoexcept

获取反卷积的多维填充。

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

另请参阅: setPaddingNd()

◆ getPostPadding()

Dims nvinfer1::IDeconvolutionLayer::getPostPadding ( ) const

inlinenoexcept

获取后填充。

另请参阅: setPostPadding()

◆ getPrePadding()

Dims nvinfer1::IDeconvolutionLayer::getPrePadding ( ) const

inlinenoexcept

获取前填充。

另请参阅: setPrePadding()

◆ getStrideNd()

Dims nvinfer1::IDeconvolutionLayer::getStrideNd ( ) const

inlinenoexcept

获取反卷积的多维步长。

另请参阅: setStrideNd()

◆ setBiasWeights()

void nvinfer1::IDeconvolutionLayer::setBiasWeights ( Weights weights )

inlinenoexcept

设置反卷积的偏置权重。

偏置是可选的。要省略偏置，请将权重结构的计数设置为零。

偏置是按特征图应用的，因此权重数（如果非零）必须等于输出特征图的数量。

另请参阅: getBiasWeights()

◆ setDilationNd()

void nvinfer1::IDeconvolutionLayer::setDilationNd ( Dims const & dilation )

inlinenoexcept

设置反卷积的多维膨胀率。

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

另请参阅: getDilationNd()

◆ setInput()

void nvinfer1::ILayer::setInput	(	int32_t	index,
		ITensor &	tensor
	)

inlinenoexcept

使用指定的张量追加或替换此层的输入。

参数

index	index
tensor	新的输入张量

输入 0 是输入激活张量。输入 1 是内核张量。如果使用，则内核权重参数必须设置为空权重。输入 2 是偏置张量。如果使用，则偏置参数必须设置为空权重。

另请参阅: getKernelWeights(), setKernelWeights(), getBiasWeights(), setBiasWeights()

◆ setKernelSizeNd()

void nvinfer1::IDeconvolutionLayer::setKernelSizeNd ( Dims const & kernelSize )

inlinenoexcept

设置反卷积的多维内核大小。

如果在 DLA 上执行此层，则有两个限制：1) 仅支持 2D 内核。2) 内核高度和宽度必须在 [1,32] 范围内，或者在一维中为 [64, 96, 128] 且在另一维中为 1 的组合，即 [1x64] 或 [64x1] 是有效的，但 [64x64] 无效。

另请参阅: getKernelSizeNd()

◆ setKernelWeights()

void nvinfer1::IDeconvolutionLayer::setKernelWeights ( Weights weights )

inlinenoexcept

设置反卷积的内核权重。

权重被指定为 CKRS 顺序的连续数组，其中 C 是输入通道数，K 是输出特征图数，R 和 S 是滤波器的高度和宽度。

另请参阅: getWeights()

◆ setNbGroups()

void nvinfer1::IDeconvolutionLayer::setNbGroups ( int64_t nbGroups )

inlinenoexcept

设置反卷积的分组数。

输入张量通道被分成 nbGroups 组，并且为每个组执行反卷积，每个组使用一个滤波器。组卷积的结果被连接起来以形成输出。

如果在 DLA 上执行此层，则 nbGroups 必须为 1

注意: 当在 int8 模式下使用组时，组的大小（即通道数除以组数）对于输入和输出都必须是 4 的倍数。

默认值：1

另请参阅: getNbGroups()

◆ setNbOutputMaps()

void nvinfer1::IDeconvolutionLayer::setNbOutputMaps ( int64_t nbOutputMaps )

inlinenoexcept

设置反卷积的输出特征图数量。

如果在 DLA 上执行此层，则输出图的数量必须在 [1,8192] 范围内。

另请参阅: getNbOutputMaps()

◆ setPaddingMode()

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

inlinenoexcept

设置填充模式。

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

默认值：kEXPLICIT_ROUND_DOWN

另请参阅: getPaddingMode()

◆ setPaddingNd()

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

inlinenoexcept

设置反卷积的多维填充。

输出将在每个维度的两侧被修剪掉此数量的元素。换句话说，它类似于具有此填充大小的反卷积层的逆运算。填充是对称的，不支持负填充。

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

如果在 DLA 上执行此层，则填充必须为 0。

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

◆ setPostPadding()

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

inlinenoexcept

设置反卷积的多维后填充。

输出将在每个维度的末尾被修剪掉此数量的元素。换句话说，它类似于具有此填充大小的反卷积层的逆运算。不支持负填充。

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

另请参阅: getPostPadding()

◆ setPrePadding()

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

inlinenoexcept

设置反卷积的多维前填充。

输出将在每个维度的开始处被修剪掉此数量的元素。换句话说，它类似于具有此填充大小的反卷积层的逆运算。不支持负填充。

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

另请参阅: getPrePadding()

◆ setStrideNd()

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

inlinenoexcept

设置反卷积的多维步长。

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

如果在 DLA 上执行此层，则有两个限制：1) 仅支持 2D 步长。2) 步长高度和宽度必须在 [1,32] 范围内，或者在一维中为 [64, 96, 128] 且在另一维中为 1 的组合，即 [1x64] 或 [64x1] 是有效的，但 [64x64] 无效。

另请参阅: getStrideNd()

成员数据文档

◆ mImpl

apiv::VDeconvolutionLayer* nvinfer1::IDeconvolutionLayer::mImpl

protected

此类文档由以下文件生成：

NvInfer.h

公共成员函数

保护成员函数

保护属性

详细描述

构造函数 & 析构函数文档

◆ ~IDeconvolutionLayer()

成员函数文档

◆ getBiasWeights()

◆ getDilationNd()

◆ getKernelSizeNd()

◆ getKernelWeights()

◆ getNbGroups()

◆ getNbOutputMaps()

◆ getPaddingMode()

◆ getPaddingNd()

◆ getPostPadding()

◆ getPrePadding()

◆ getStrideNd()

◆ setBiasWeights()

◆ setDilationNd()

◆ setInput()

◆ setKernelSizeNd()

◆ setKernelWeights()

◆ setNbGroups()

◆ setNbOutputMaps()

◆ setPaddingMode()

◆ setPaddingNd()

◆ setPostPadding()

◆ setPrePadding()

◆ setStrideNd()

成员数据文档

◆ mImpl