在TensorRT8中,对张量的维度表示数据结构略有修改。主要是新增了Dims32,原本的Dims现在仅是Dims32的别称。同时废弃了DimsCHW和DimsNCHW[1]

TensorRT8 Dim32 继承关系(一)

TensorRT8 Dim32 继承关系(二)

TensorRT6 and 7 Dims 继承关系
| Descriptor | 功能 | 8 | 7 | 6 |
|---|---|---|---|---|
| Dims | 多维数据 | Y | Y | Y |
| Dims2 | 二维数据 | Y | Y | Y |
| Dims3 | 三维数据 | Y | Y | Y |
| Dims32[1] | 多维数据 | Y | 不支持 | 不支持 |
| DimsHW[2] | 二维空间数据 | Y | Y | Y |
| DimsCHW | 通道+二维空间数据 | 已废弃 | Y | Y |
| Dims4 | 四维数据 | Y | Y | Y |
| DimsNCHW | 索引+通道+二维空间数据 | 已废弃 | Y | Y |