nvinfer1::IActivationLayer 类参考

网络定义中的激活层。 更多...

#include <NvInfer.h>

nvinfer1::IActivationLayer 的继承关系图

公共成员函数
void	setActivationType (ActivationType type) noexcept
	设置要执行的激活类型。 更多...
ActivationType	getActivationType () const noexcept
	获取要执行的激活类型。 更多...
void	setAlpha (float alpha) noexcept
	设置 alpha 参数（必须是有限值）。 更多...
void	setBeta (float beta) noexcept
	设置 beta 参数（必须是有限值）。 更多...
float	getAlpha () const noexcept
	获取 alpha 参数。 更多...
float	getBeta () const noexcept
	获取 beta 参数。 更多...
继承自 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	~IActivationLayer () 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::VActivationLayer *	mImpl
继承自 nvinfer1::ILayer 的受保护的属性
apiv::VLayer *	mLayer

详细描述

网络定义中的激活层。

此层对其输入应用逐元素激活函数。

输出与输入具有相同的形状。

如果输出是形状张量，则输入是形状张量。

警告

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

构造函数 & 析构函数文档

~IActivationLayer()

virtual nvinfer1::IActivationLayer::~IActivationLayer ( )

protectedvirtualdefaultnoexcept

成员函数文档

getActivationType()

ActivationType nvinfer1::IActivationLayer::getActivationType ( ) const

inlinenoexcept

获取要执行的激活类型。

另请参阅

setActivationType(), ActivationType

getAlpha()

float nvinfer1::IActivationLayer::getAlpha ( ) const

inlinenoexcept

获取 alpha 参数。

另请参阅

getBeta(), setAlpha()

getBeta()

float nvinfer1::IActivationLayer::getBeta ( ) const

inlinenoexcept

获取 beta 参数。

另请参阅

getAlpha(), setBeta()

setActivationType()

void nvinfer1::IActivationLayer::setActivationType ( ActivationType  type )

inlinenoexcept

设置要执行的激活类型。

在 DLA 上，有效的激活类型为 kRELU、kSIGMOID、kTANH 和 kCLIP。

另请参阅

getActivationType(), ActivationType

setAlpha()

void nvinfer1::IActivationLayer::setAlpha ( float  alpha )

inlinenoexcept

设置 alpha 参数（必须是有限值）。

以下激活函数使用此参数：LeakyRelu、Elu、Selu、Softplus、Clip、HardSigmoid、ScaledTanh、ThresholdedRelu。

其他激活函数会忽略它。

另请参阅

getAlpha(), setBeta()

setBeta()

void nvinfer1::IActivationLayer::setBeta ( float  beta )

inlinenoexcept

设置 beta 参数（必须是有限值）。

以下激活函数使用此参数：Selu、Softplus、Clip、HardSigmoid、ScaledTanh。

其他激活函数会忽略它。

另请参阅

getBeta(), setAlpha()

成员数据文档

mImpl

apiv::VActivationLayer* nvinfer1::IActivationLayer::mImpl

protected

---

此类文档从以下文件生成

• NvInfer.h