nvinfer1::IReduceLayer 类参考

表示跨非布尔张量执行归约的层。 更多...

#include <NvInfer.h>

nvinfer1::IReduceLayer 的继承关系图

公共成员函数
void	setOperation (ReduceOperation op) noexcept
	设置层的归约操作。 更多...
ReduceOperation	getOperation () const noexcept
	获取层的归约操作。 更多...
void	setReduceAxes (uint32_t reduceAxes) noexcept
	设置要执行归约的轴。 更多...
uint32_t	getReduceAxes () const noexcept
	获取层的要执行归约的轴。 更多...
void	setKeepDimensions (bool keepDimensions) noexcept
	设置布尔值，指定是否为层保留缩减的维度。 更多...
bool	getKeepDimensions () 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	~IReduceLayer () 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::VReduceLayer *	mImpl
继承自 nvinfer1::ILayer 的保护属性
apiv::VLayer *	mLayer

详细描述

表示跨非布尔张量执行归约的层。

警告

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

构造函数和析构函数文档

~IReduceLayer()

virtual nvinfer1::IReduceLayer::~IReduceLayer ( )

protectedvirtualdefaultnoexcept

成员函数文档

getKeepDimensions()

bool nvinfer1::IReduceLayer::getKeepDimensions ( ) const

inlinenoexcept

获取布尔值，该值指定是否为层保留缩减的维度。

另请参阅

setKeepDimensions

getOperation()

ReduceOperation nvinfer1::IReduceLayer::getOperation ( ) const

inlinenoexcept

获取层的归约操作。

另请参阅

setOperation(), ReduceOperation

getReduceAxes()

uint32_t nvinfer1::IReduceLayer::getReduceAxes ( ) const

inlinenoexcept

获取层的要执行归约的轴。

另请参阅

setReduceAxes

setKeepDimensions()

void nvinfer1::IReduceLayer::setKeepDimensions ( bool  keepDimensions )

inlinenoexcept

设置布尔值，指定是否为层保留缩减的维度。

另请参阅

getKeepDimensions

setOperation()

void nvinfer1::IReduceLayer::setOperation ( ReduceOperation  op )

inlinenoexcept

设置层的归约操作。

另请参阅

getOperation(), ReduceOperation

setReduceAxes()

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

inlinenoexcept

设置要执行归约的轴。

另请参阅

getReduceAxes

成员数据文档

mImpl

apiv::VReduceLayer* nvinfer1::IReduceLayer::mImpl

protected

---

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

• NvInfer.h