表示矩阵乘法的层。更多...

#include <NvInfer.h>

nvinfer1::IMatrixMultiplyLayer 的继承关系图

公共成员函数
void	setOperation (int32_t index, MatrixOperation op) noexcept
	设置输入张量的操作。更多...

MatrixOperation	getOperation (int32_t index) 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	~IMatrixMultiplyLayer () 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::VMatrixMultiplyLayer *	mImpl

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

详细描述

表示矩阵乘法的层。

设 A 为 op(getInput(0))，B 为 op(getInput(1))，其中 op(x) 表示对应的 MatrixOperation。

当 A 和 B 是矩阵或向量时，计算内积 A * B

matrix * matrix -> matrix
matrix * vector -> vector
vector * matrix -> vector
vector * vector -> scalar

更高秩的输入被视为矩阵或向量的集合。输出将是相应的矩阵、向量或标量的集合。

对于不是矩阵或向量维度的维度：如果其中一个张量的维度为 1，而另一个张量不是，则前一个张量将沿该维度广播以匹配后一个张量的维度。A 和 B 的这些额外维度的数量必须匹配。

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

构造函数 & 析构函数文档

◆ ~IMatrixMultiplyLayer()

virtual nvinfer1::IMatrixMultiplyLayer::~IMatrixMultiplyLayer ( )

protectedvirtualdefaultnoexcept

成员函数文档

◆ getOperation()

MatrixOperation nvinfer1::IMatrixMultiplyLayer::getOperation ( int32_t index ) const

inlinenoexcept

获取输入张量的操作。

参数

index 输入张量编号 (0 或 1)。

参见: setOperation()

◆ setOperation()

void nvinfer1::IMatrixMultiplyLayer::setOperation	(	int32_t	index,
		MatrixOperation	op
	)

inlinenoexcept

设置输入张量的操作。

参数

index	输入张量编号 (0 或 1)。
op	新操作。

参见: getOperation()

成员数据文档

◆ mImpl

apiv::VMatrixMultiplyLayer* nvinfer1::IMatrixMultiplyLayer::mImpl

protected

此类文档从以下文件生成

NvInfer.h

公共成员函数

保护成员函数

保护属性