nvinfer1::IConcatenationLayer Class Reference

A concatenation layer in a network definition. More...

#include <NvInfer.h>

Inheritance diagram for nvinfer1::IConcatenationLayer:

Public Member Functions
virtual void	setAxis (int axis)=0
	Set the axis along which concatenation occurs. More...
virtual int	getAxis () const =0
	Get the axis along which concatenation occurs. 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...

Detailed Description

A concatenation layer in a network definition.

The output channel size is the sum of the channel sizes of the inputs. The other output sizes are the same as the other input sizes, which must all match.

Member Function Documentation

virtual int nvinfer1::IConcatenationLayer::getAxis ( ) const

pure virtual

Get the axis along which concatenation occurs.

See Also

setAxis()

virtual void nvinfer1::IConcatenationLayer::setAxis ( int  axis )

pure virtual

Set the axis along which concatenation occurs.

0 is the major axis (excluding the batch dimension). The default is the number of non-batch axes in the tensor minus three (e.g. for an NCHW input it would be 0), or 0 if there are fewer than 3 non-batch axes.

Parameters

axis	The axis along which concatenation occurs.

---

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

• NvInfer.h