nvinfer1::IMatrixMultiplyLayer Class Reference

Layer that represents a Matrix Multiplication. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IMatrixMultiplyLayer:

Public Member Functions
virtual void	setOperation (int index, MatrixOperation op)=0
	Set the operation for an input tensor. More...
virtual MatrixOperation	getOperation (int index) const =0
	Get the operation for an input tensor. More...
virtual TRT_DEPRECATED void	setTranspose (int index, bool val)=0
	Set the transpose flag for an input tensor. More...
virtual TRT_DEPRECATED bool	getTranspose (int index) const =0
	Get the transpose flag for an input tensor. More...
Public Member Functions inherited from nvinfer1::ILayer
virtual LayerType	getType () const =0
	Return the type of a layer. More...
virtual void	setName (const char *name)=0
	Set the name of a layer. More...
virtual const char *	getName () const =0
	Return the name of a layer. More...
virtual int	getNbInputs () const =0
	Get the number of inputs of a layer.
virtual ITensor *	getInput (int index) const =0
	Get the layer input corresponding to the given index. More...
virtual int	getNbOutputs () const =0
	Get the number of outputs of a layer.
virtual ITensor *	getOutput (int index) const =0
	Get the layer output corresponding to the given index. More...
virtual void	setInput (int index, ITensor &tensor)=0
	replace an input of this layer with a specific tensor More...
virtual void	setPrecision (DataType dataType)=0
	Set the computational precision of this layer. More...
virtual DataType	getPrecision () const =0
	get the computational precision of this layer More...
virtual bool	precisionIsSet () const =0
	whether the computational precision has been set for this layer More...
virtual void	resetPrecision ()=0
	reset the computational precision for this layer More...
virtual void	setOutputType (int index, DataType dataType)=0
	Set the output type of this layer. More...
virtual DataType	getOutputType (int index) const =0
	get the output type of this layer More...
virtual bool	outputTypeIsSet (int index) const =0
	whether the output type has been set for this layer More...
virtual void	resetOutputType (int index)=0
	reset the output type for this layer More...

Detailed Description

Layer that represents a Matrix Multiplication.

Let A be op(getInput(0)) and B be op(getInput(1)) where op(x) denotes the corresponding MatrixOperation.

When A and B are matrices or vectors, computes the inner product A * B:

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

Inputs of higher rank are treated as collections of matrices or vectors. The output will be a corresponding collection of matrices, vectors, or scalars.

For a dimension that is not one of the matrix or vector dimensions: If the dimension is 1 for one of the tensors but not the other tensor, the former tensor is broadcast along that dimension to match the dimension of the latter tensor. The number of these extra dimensions for A and B must match.

Warning

Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.

Member Function Documentation

getOperation()

virtual MatrixOperation nvinfer1::IMatrixMultiplyLayer::getOperation ( int  index ) const

pure virtual

Get the operation for an input tensor.

Parameters

index

Input tensor number (0 or 1).

See also

setTranspose()

getTranspose()

virtual TRT_DEPRECATED bool nvinfer1::IMatrixMultiplyLayer::getTranspose ( int  index ) const

pure virtual

Get the transpose flag for an input tensor.

Parameters

index

Input tensor number (0 or 1).

See also

setTranspose()

Deprecated:

getTranspose is superseded by getOperation.

setOperation()

virtual void nvinfer1::IMatrixMultiplyLayer::setOperation ( int  index, MatrixOperation  op  )

pure virtual

Set the operation for an input tensor.

Parameters

index	Input tensor number (0 or 1).
op	New operation.

See also

getTranspose()

setTranspose()

virtual TRT_DEPRECATED void nvinfer1::IMatrixMultiplyLayer::setTranspose ( int  index, bool  val  )

pure virtual

Set the transpose flag for an input tensor.

Parameters

index	Input tensor number (0 or 1).
val	New transpose flag.

See also

getTranspose()

Deprecated:

setTranspose is superseded by setOperation.

---

The documentation for this class was generated from the following file:

• NvInfer.h