网络中的 Einsum 层。更多...

#include <NvInfer.h>

nvinfer1::IEinsumLayer 的继承关系图

公共成员函数
bool	setEquation (char const *equation) noexcept
	设置方程。方程是以逗号分隔的下标标签列表，其中每个标签引用相应张量的维度。更多...

char const *	getEquation () 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	~IEinsumLayer () 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::VEinsumLayer *	mImpl

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

详细描述

网络中的 Einsum 层。

此层根据爱因斯坦求和约定，对输入元素沿由 equation 参数指定的维度进行求和。该层可以有一个或多个 rank >= 0 的输入。所有输入必须具有 DataType::kFLOAT 或 DataType::kHALF 类型，不一定相同。有一个类型为 DataType::kFLOAT 的输出。输出张量的形状由方程确定。

方程为每个输入中的每个维度指定 ASCII 小写字母，维度顺序与输入顺序相同，每个输入之间用逗号分隔。用相同下标标记的维度必须匹配或可广播。一个输入中重复的下标标签取对角线。跨多个输入重复标签意味着这些轴将被相乘。从输出中省略标签意味着将对沿这些轴的值求和。在隐式模式下，表达式中出现一次的索引将按字母顺序递增成为输出的一部分。在显式模式下，可以通过添加箭头 ('->') 后跟输出的下标来控制输出。例如，“ij,jk->ik” 等价于 “ij,jk”。省略号 ('...') 可以用来代替下标来广播维度。有关方程语法的更多详细信息，请参阅 TensorRT 开发者指南。

许多常见操作可以使用 Einsum 方程表示。例如：矩阵转置：ij->ji 求和：ij-> 矩阵-矩阵乘法：ik,kj->ij 点积：i,i-> 矩阵-向量乘法：ik,k->i 批矩阵乘法：ijk,ikl->ijl 批对角线：...ii->...i

注意: TensorRT 不支持省略号、对角线操作或 Einsum 的两个以上输入。

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

构造函数和析构函数文档

◆ ~IEinsumLayer()

virtual nvinfer1::IEinsumLayer::~IEinsumLayer ( )

protectedvirtualdefaultnoexcept

成员函数文档

◆ getEquation()

char const * nvinfer1::IEinsumLayer::getEquation ( ) const

inlinenoexcept

返回方程。

另请参阅: setEquation()

◆ setEquation()

bool nvinfer1::IEinsumLayer::setEquation ( char const * equation )

inlinenoexcept

设置方程。方程是以逗号分隔的下标标签列表，其中每个标签引用相应张量的维度。

返回值: 如果方程在语法上有效并成功设置，则返回 true，否则返回 false。

另请参阅: setEquation()

成员数据文档

◆ mImpl

apiv::VEinsumLayer* nvinfer1::IEinsumLayer::mImpl

protected

此类文档从以下文件生成

NvInfer.h

公共成员函数

保护成员函数

保护属性