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