nvinfer1::IPaddingLayer 类参考

表示填充操作的层。更多...

#include <NvInfer.h>

nvinfer1::IPaddingLayer 的继承关系图

公共成员函数
void	setPrePaddingNd (Dims const &padding) noexcept
	设置在张量开始处应用的填充。更多...
Dims	getPrePaddingNd () const noexcept
	获取在张量开始处应用的填充。更多...
void	setPostPaddingNd (Dims const &padding) noexcept
	设置在张量末尾应用的填充。更多...
Dims	getPostPaddingNd () 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	~IPaddingLayer () 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::VPaddingLayer *	mImpl
继承自 nvinfer1::ILayer 的保护属性
apiv::VLayer *	mLayer

详细描述

表示填充操作的层。

填充层在输入张量的开始和结束处添加零填充。它仅支持填充最后两个维度。应用负填充会导致裁剪输入。

要跨任何维度子集进行填充，请使用 ISliceLayer 和 SampleMode::kFILL。

警告

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

构造函数 & 析构函数文档

~IPaddingLayer()

virtual nvinfer1::IPaddingLayer::~IPaddingLayer ( )

protectedvirtualdefaultnoexcept

成员函数文档

getPostPaddingNd()

Dims nvinfer1::IPaddingLayer::getPostPaddingNd ( ) const

inlinenoexcept

获取在张量末尾应用的填充。

警告

目前仅支持 2 维填充。

参见

setPostPaddingNd

getPrePaddingNd()

Dims nvinfer1::IPaddingLayer::getPrePaddingNd ( ) const

inlinenoexcept

获取在张量开始处应用的填充。

警告

目前仅支持 2 维填充。

参见

setPrePaddingNd

setPostPaddingNd()

void nvinfer1::IPaddingLayer::setPostPaddingNd ( Dims const &  padding )

inlinenoexcept

设置在张量末尾应用的填充。

负填充会导致按指定量修剪边缘。

警告

目前仅支持 2 维填充。

参见

getPostPaddingNd

setPrePaddingNd()

void nvinfer1::IPaddingLayer::setPrePaddingNd ( Dims const &  padding )

inlinenoexcept

设置在张量开始处应用的填充。

负填充会导致按指定量修剪边缘。

警告

目前仅支持 2 维填充。

参见

getPrePaddingNd

成员数据文档

mImpl

apiv::VPaddingLayer* nvinfer1::IPaddingLayer::mImpl

protected

---

此类的文档从以下文件生成：

• NvInfer.h