nvinfer1::Dims Class Reference
Structure to define the dimensions of a tensor. More...
#include <NvInferRuntimeCommon.h>
Inheritance diagram for nvinfer1::Dims:
Public Attributes | |
| int32_t | nbDims |
| The number of dimensions. | |
| int32_t | d [MAX_DIMS] |
| The extent of each dimension. | |
Static Public Attributes | |
| static constexpr int32_t | MAX_DIMS {8} |
| The maximum number of dimensions supported for a tensor. | |
Detailed Description
Structure to define the dimensions of a tensor.
TensorRT can also return an invalid dims structure. This structure is represented by nbDims == -1 and d[i] == 0 for all d.
TensorRT can also return an "unknown rank" dims structure. This structure is represented by nbDims == -1 and d[i] == -1 for all d.
The documentation for this class was generated from the following file:
