nvinfer1::IIteratorLayer 类参考

用于执行迭代的层。 更多...

#include <NvInfer.h>

nvinfer1::IIteratorLayer 的继承关系图

公共成员函数
void	setAxis (int32_t axis) noexcept
	设置要迭代的轴。 更多...
int32_t	getAxis () const noexcept
	获取正在迭代的轴。 更多...
void	setReverse (bool reverse) noexcept
	设置迭代顺序为反向。 更多...
bool	getReverse () const noexcept
	检查迭代顺序是否为反向。 更多...
继承自 nvinfer1::ILoopBoundaryLayer 的公共成员函数
ILoop *	getLoop () const noexcept
	获取指向与此边界层关联的 ILoop 的指针。 更多...
继承自 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	~IIteratorLayer () noexcept=default
继承自 nvinfer1::ILoopBoundaryLayer 的保护成员函数
virtual	~ILoopBoundaryLayer () 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::VIteratorLayer *	mImpl
继承自 nvinfer1::ILoopBoundaryLayer 的保护属性
apiv::VLoopBoundaryLayer *	mBoundary
继承自 nvinfer1::ILayer 的保护属性
apiv::VLayer *	mLayer

详细描述

用于执行迭代的层。

迭代器层沿着给定的轴和给定的方向迭代张量。 它使每次循环迭代都可以检查张量的不同切片。

参见

ILoop::addIterator()

构造函数 & 析构函数文档

~IIteratorLayer()

virtual nvinfer1::IIteratorLayer::~IIteratorLayer ( )

protectedvirtualdefaultnoexcept

成员函数文档

getAxis()

int32_t nvinfer1::IIteratorLayer::getAxis ( ) const

inlinenoexcept

获取正在迭代的轴。

getReverse()

bool nvinfer1::IIteratorLayer::getReverse ( ) const

inlinenoexcept

检查迭代顺序是否为反向。

返回值

当且仅当反转输入时为 True。

setAxis()

void nvinfer1::IIteratorLayer::setAxis ( int32_t  axis )

inlinenoexcept

设置要迭代的轴。

setReverse()

void nvinfer1::IIteratorLayer::setReverse ( bool  reverse )

inlinenoexcept

设置迭代顺序为反向。

对于 reverse=false，该层等效于 addGather(tensor, I, 0)，其中 I 是包含循环迭代次数的标量张量。 对于 reverse=true，该层等效于 addGather(tensor, M-1-I, 0)，其中 M 是从 kCOUNT 类型的 TripLimits 计算出的行程计数。 默认值为 reverse=false。

成员数据文档

mImpl

apiv::VIteratorLayer* nvinfer1::IIteratorLayer::mImpl

protected

---

此类文档从以下文件生成

• NvInfer.h