TensorRT
8.0.0
|
Structure to define the dimensions of a tensor. More...
#include <NvInferRuntimeCommon.h>
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. | |
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.