nvinfer1::ITopKLayer 类参考

表示 TopK 缩减的层。更多...

#include <NvInfer.h>

nvinfer1::ITopKLayer 的继承关系图

公共成员函数
void	setOperation (TopKOperation op) noexcept
	设置层的操作。更多...
TopKOperation	getOperation () const noexcept
	获取层的操作。更多...
void	setK (int32_t k) noexcept
	设置层的静态 k 值。更多...
int32_t	getK () const noexcept
	获取层的 k 值。更多...
void	setReduceAxes (uint32_t reduceAxes) noexcept
	设置层要缩减的轴。更多...
uint32_t	getReduceAxes () 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	~ITopKLayer () 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::VTopKLayer *	mImpl
继承自 nvinfer1::ILayer 的受保护属性
apiv::VLayer *	mLayer

详细描述

表示 TopK 缩减的层。

此层可以接受静态和动态 k 值。静态 k 值可以通过 addTopK() API 函数设置，或在使用层创建后的 getK() 和 setK() 函数访问。对于动态 k 值，请使用 setInput() 方法传入 k 值作为索引为 1 的张量，这将覆盖计算中的静态 k 值。

警告

请勿从此类继承，因为这样做会破坏 API 和 ABI 的向前兼容性。

构造函数 & 析构函数文档

~ITopKLayer()

virtual nvinfer1::ITopKLayer::~ITopKLayer ( )

protectedvirtualdefaultnoexcept

成员函数文档

getK()

int32_t nvinfer1::ITopKLayer::getK ( ) const

inlinenoexcept

获取层的 k 值。

此函数将返回传递给 addTopK() 的静态 k 值，或传递给 setK() 的值。

如果存在第二个层输入且非空，则此函数返回 -1。

参见

setK()

getOperation()

TopKOperation nvinfer1::ITopKLayer::getOperation ( ) const

inlinenoexcept

获取层的操作。

参见

setOperation(), TopKOperation

getReduceAxes()

uint32_t nvinfer1::ITopKLayer::getReduceAxes ( ) const

inlinenoexcept

获取层要缩减的轴。

参见

setReduceAxes()

setInput()

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

inlinenoexcept

为此层附加或替换特定的输入张量。

参数

index	要修改的输入的索引。
tensor	新的输入张量。

对于 TopK 层，有效值为 0-1。索引如下：

• 0：输入数据张量。
• 1：一个标量 Int32 张量，包含一个正值，对应于要检索的顶部元素的数量。大于 3840 的值将导致运行时错误。如果提供，这将覆盖计算中的静态 k 值。

setK()

void nvinfer1::ITopKLayer::setK ( int32_t  k )

inlinenoexcept

设置层的静态 k 值。

当前仅支持高达 3840 的值。

如果已为此层设置了第二个输入，则此方法会将其重置为空。

参见

getK()

setOperation()

void nvinfer1::ITopKLayer::setOperation ( TopKOperation  op )

inlinenoexcept

设置层的操作。

参见

getOperation(), TopKOperation

setReduceAxes()

void nvinfer1::ITopKLayer::setReduceAxes ( uint32_t  reduceAxes )

inlinenoexcept

设置层要缩减的轴。

参见

getReduceAxes()

成员数据文档

mImpl

apiv::VTopKLayer* nvinfer1::ITopKLayer::mImpl

protected

---

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

• NvInfer.h