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::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.
 
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: