TensorRT
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Pages
nvinfer1::Dims Class Reference

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

#include <NvInfer.h>

Inheritance diagram for nvinfer1::Dims:
nvinfer1::DimsCHW nvinfer1::DimsHW nvinfer1::DimsNCHW

Public Attributes

int nbDims
 the number of dimensions
 
int d [MAX_DIMS]
 the extent of each dimension
 
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)

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