nvinfer1::IIfConditional 类参考

用于构建条件执行子图的辅助类。 更多...

#include <NvInfer.h>

nvinfer1::IIfConditional 的继承关系图

公共成员函数
IConditionLayer *	setCondition (ITensor &condition) noexcept
	为此 If-Conditional 结构设置条件张量。 更多...
IIfConditionalOutputLayer *	addOutput (ITensor &trueSubgraphOutput, ITensor &falseSubgraphOutput) noexcept
	添加 If-conditional 输出。 更多...
IIfConditionalInputLayer *	addInput (ITensor &input) noexcept
	添加 If-conditional 输入。 更多...
void	setName (char const *name) noexcept
	设置 conditional 的名称。 更多...
char const *	getName () const noexcept
	返回 conditional 的名称。 更多...

保护成员函数
virtual	~IIfConditional () 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::VIfConditional *	mImpl

详细描述

用于构建条件执行子图的辅助类。

If-conditional 根据以下伪代码有条件地执行网络的一部分

如果 condition 为真: output = trueSubgraph(trueInputs); 否则 output = falseSubgraph(falseInputs); 发射 output

Condition 是一个 0D 布尔张量（表示标量）。 trueSubgraph 表示当 condition 评估为 True 时执行的网络子图。 falseSubgraph 表示当 condition 评估为 False 时执行的网络子图。

以下约束适用于 If-conditionals

• trueSubgraph 和 falseSubgraph 都必须被定义。
• 两个子图中的输出张量数量必须相同。
• 来自 true/false 子图的对应输出张量必须具有相同的类型和形状。

构造函数 & 析构函数 文档

~IIfConditional()

virtual nvinfer1::IIfConditional::~IIfConditional ( )

protectedvirtualdefaultnoexcept

成员函数文档

addInput()

IIfConditionalInputLayer * nvinfer1::IIfConditional::addInput ( ITensor &  input )

inlinenoexcept

添加 If-conditional 输入。

参数

input

conditional 的一个输入，可以被 conditional 的子图中的一个或两个使用。

另请参阅

IIfConditionalInputLayer

addOutput()

IIfConditionalOutputLayer * nvinfer1::IIfConditional::addOutput ( ITensor &  trueSubgraphOutput, ITensor &  falseSubgraphOutput  )

inlinenoexcept

添加 If-conditional 输出。

参数

trueSubgraphOutput	当 conditional 评估为 true 时执行的子图的输出。
falseSubgraphOutput	当 conditional 评估为 false 时执行的子图的输出。

IIfConditional 的每个输出层代表 IIfConditional 的 true-subgraph 或 false-subgraph 的单个输出，具体取决于执行哪个子图。

除非 condition 是构建时常量，否则两个张量的形状必须相等。

另请参阅

IIfConditionalOutputLayer

getName()

char const * nvinfer1::IIfConditional::getName ( ) const

inlinenoexcept

返回 conditional 的名称。

另请参阅

setName()

setCondition()

IConditionLayer * nvinfer1::IIfConditional::setCondition ( ITensor &  condition )

inlinenoexcept

为此 If-Conditional 结构设置条件张量。

参数

condition

将确定执行哪个子图的条件张量。

condition 张量必须是类型为 DataType::kBOOL 的 0D 执行张量（标量）。

另请参阅

IConditionLayer

setName()

void nvinfer1::IIfConditional::setName ( char const *  name )

inlinenoexcept

设置 conditional 的名称。

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

警告

字符串名称必须以 null 结尾，并且最多 4096 字节，包括终止符。

另请参阅

getName()

成员数据文档

mImpl

apiv::VIfConditional* nvinfer1::IIfConditional::mImpl

protected

---

此类文档从以下文件生成

• NvInfer.h