nvinfer1::IShapeLayer Class Referenceabstract
Layer type for getting shape of a tensor. More...
#include <NvInfer.h>
Inheritance diagram for nvinfer1::IShapeLayer:
Protected Member Functions | |
| virtual | ~IShapeLayer () noexcept=0 |
 Protected Member Functions inherited from nvinfer1::ILayer | |
| virtual | ~ILayer () noexcept=0 |
| Protected Member Functions inherited from 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 |
Protected Attributes | |
| apiv::VShapeLayer * | mImpl |
| Protected Attributes inherited from nvinfer1::ILayer | |
| apiv::VLayer * | mLayer |
Additional Inherited Members | |
| Public Member Functions inherited from nvinfer1::ILayer | |
| LayerType | getType () const noexcept |
| Return the type of a layer. More... | |
| void | setName (char const *name) noexcept |
| Set the name of a layer. More... | |
| char const * | getName () const noexcept |
| Return the name of a layer. More... | |
| int32_t | getNbInputs () const noexcept |
| Get the number of inputs of a layer. More... | |
| ITensor * | getInput (int32_t index) const noexcept |
| Get the layer input corresponding to the given index. More... | |
| int32_t | getNbOutputs () const noexcept |
| Get the number of outputs of a layer. More... | |
| ITensor * | getOutput (int32_t index) const noexcept |
| Get the layer output corresponding to the given index. More... | |
| void | setInput (int32_t index, ITensor &tensor) noexcept |
| Replace an input of this layer with a specific tensor. More... | |
| DataType | getOutputType (int32_t index) const noexcept |
| get the output type of this layer More... | |
| void | setMetadata (char const *metadata) noexcept |
| Set the metadata for this layer. More... | |
| char const * | getMetadata () const noexcept |
| Get the metadata of the layer. More... | |
| bool | setNbRanks (int32_t nbRanks) noexcept |
| Set the number of ranks for multi-device execution. More... | |
| int32_t | getNbRanks () const noexcept |
| Get the number of ranks for multi-device execution. More... | |
Detailed Description
Layer type for getting shape of a tensor.
This layer sets the output to a 1D tensor of type Int64 with the dimensions of the input tensor.
For example, if the input is a four-dimensional tensor (of any type) with dimensions [2,3,5,7], the output tensor is a one-dimensional Int64 tensor of length 4 containing the sequence 2, 3, 5, 7.
- Warning
- Do not inherit from this class, as doing so will break forward-compatibility of the API and ABI.
Constructor & Destructor Documentation
◆ ~IShapeLayer()
|
inlineprotectedpure virtualdefaultnoexcept |
Member Data Documentation
◆ mImpl
|
protected |
The documentation for this class was generated from the following file:
