网络中的断言层。更多...

#include <NvInfer.h>

nvinfer1::IAssertionLayer 的继承关系图

公共成员函数
void	setMessage (char const *message) noexcept
	设置断言失败时打印的消息。更多...

char const *	getMessage () 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	~IAssertionLayer () 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::VAssertionLayer *	mImpl

继承自 nvinfer1::ILayer 的保护属性
apiv::VLayer *	mLayer

详细描述

网络中的断言层。

该层具有单个输入且没有输出。输入必须是布尔形状张量。如果在构建时输入中的任何元素被证明为假，则网络将被拒绝。如果在运行时为提供的运行时维度输入中的任何元素为假，则会发生错误，就像任何其他运行时错误（例如，使用 IShuffleLayer 更改张量的体积）被处理一样。

断言输入维度的相等性可能有助于优化器。

警告: 不要从此类继承，因为这样做会破坏 API 和 ABI 的向前兼容性。

构造函数 & 析构函数文档

◆ ~IAssertionLayer()

virtual nvinfer1::IAssertionLayer::~IAssertionLayer ( )

protectedvirtualdefaultnoexcept

成员函数文档

◆ getMessage()

char const * nvinfer1::IAssertionLayer::getMessage ( ) const

inlinenoexcept

返回断言消息。

另请参阅: setMessage()

◆ setMessage()

void nvinfer1::IAssertionLayer::setMessage ( char const * message )

inlinenoexcept

设置断言失败时打印的消息。

该名称用于错误诊断。此方法复制消息字符串。

另请参阅: getMessage()

成员数据文档

◆ mImpl

apiv::VAssertionLayer* nvinfer1::IAssertionLayer::mImpl

protected

此类文档从以下文件生成

NvInfer.h

公共成员函数

保护成员函数

保护属性