nvinfer1::IConstantLayer 类参考

表示常量值的层。更多...

#include <NvInfer.h>

nvinfer1::IConstantLayer 的继承关系图

公共成员函数
void	setWeights (Weights weights) noexcept
	设置层的权重。更多...
Weights	getWeights () const noexcept
	获取层的权重。更多...
void	setDimensions (Dims const &dimensions) noexcept
	设置层的维度。更多...
Dims	getDimensions () 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	~IConstantLayer () 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::VConstantLayer *	mImpl
继承自 nvinfer1::ILayer 的受保护的属性
apiv::VLayer *	mLayer

详细描述

表示常量值的层。

注意

此层不支持布尔类型。

警告

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

构造函数 & 析构函数文档

~IConstantLayer()

virtual nvinfer1::IConstantLayer::~IConstantLayer ( )

protectedvirtualdefaultnoexcept

成员函数文档

getDimensions()

Dims nvinfer1::IConstantLayer::getDimensions ( ) const

inlinenoexcept

获取层的维度。

返回值

层的维度

参见

getDimensions

getWeights()

Weights nvinfer1::IConstantLayer::getWeights ( ) const

inlinenoexcept

获取层的权重。

参见

setWeights

setDimensions()

void nvinfer1::IConstantLayer::setDimensions ( Dims const &  dimensions )

inlinenoexcept

设置层的维度。

参数

dimensions

层的维度

参见

setDimensions

setWeights()

void nvinfer1::IConstantLayer::setWeights ( Weights  weights )

inlinenoexcept

设置层的权重。

输出类型为 weights.type。如果网络是弱类型的，并且权重具有实数类型，则根据 TensorRT 的类型转换规则，输出类型可能会有所不同。

参见

getWeights()

成员数据文档

mImpl

apiv::VConstantLayer* nvinfer1::IConstantLayer::mImpl

protected

---

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

• NvInfer.h