TensorRT  6.0.1.5
nvinfer1::Dims Class Reference

Structure to define the dimensions of a tensor. More...

#include <NvInferRuntimeCommon.h>

Inheritance diagram for nvinfer1::Dims:
nvinfer1::Dims2 nvinfer1::Dims3 nvinfer1::Dims4 nvinfer1::DimsHW nvinfer1::DimsCHW nvinfer1::DimsNCHW

Public Attributes

int nbDims
 The number of dimensions.
 
int d [MAX_DIMS]
 The extent of each dimension.
 
TRT_DEPRECATED DimensionType type [MAX_DIMS]
 The type of each dimension.
 

Static Public Attributes

static const int MAX_DIMS = 8
 The maximum number of dimensions supported for a tensor.
 

Detailed Description

Structure to define the dimensions of a tensor.

Note
: Currently the following formats are supported for layer inputs and outputs:
  • zero or more index dimensions followed by one channel and two spatial dimensions (e.g. CHW)
  • one time series dimension followed by one index dimension followed by one channel dimension (i.e. TNC)

TensorRT can also return an invalid dims structure. This structure is represented by nbDims == -1 and d[i] == 0 for all d.


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