nvinfer1::IConcatenationLayer Class Referenceabstract
A concatenation layer in a network definition. More...
#include <NvInfer.h>
Inheritance diagram for nvinfer1::IConcatenationLayer:
Public Member Functions | |
| virtual void | setAxis (int32_t axis)=0 |
| Set the axis along which concatenation occurs. More... | |
| virtual int32_t | 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 int32_t | getNbInputs () const =0 |
| Get the number of inputs of a layer. | |
| virtual ITensor * | getInput (int32_t index) const =0 |
| Get the layer input corresponding to the given index. More... | |
| virtual int32_t | getNbOutputs () const =0 |
| Get the number of outputs of a layer. | |
| virtual ITensor * | getOutput (int32_t index) const =0 |
| Get the layer output corresponding to the given index. More... | |
| virtual void | setInput (int32_t 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 (int32_t index, DataType dataType)=0 |
| Set the output type of this layer. More... | |
| virtual DataType | getOutputType (int32_t index) const =0 |
| get the output type of this layer More... | |
| virtual bool | outputTypeIsSet (int32_t index) const =0 |
| whether the output type has been set for this layer More... | |
| virtual void | resetOutputType (int32_t index)=0 |
| reset the output type for this layer 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.
- Warning
- Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.
Member Function Documentation
◆ getAxis()
|
pure virtual |
Get the axis along which concatenation occurs.
- See also
- setAxis()
◆ setAxis()
|
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.
When running this layer on the DLA, only concat across the Channel axis is valid.
- Parameters
-
axis The axis along which concatenation occurs.
The documentation for this class was generated from the following file:
