TensorRT8 使用手记(3)维度表示

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

  1. nvinfer1::Dims32 Class Reference

  2. nvinfer1::DimsHW Class Reference

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 16宿命:用概率思维提高你的胜算 以前的我是风险厌恶者,不喜欢去冒险,但是人生放弃了冒险,也就放弃了无数的可能。 ...
    yichen大刀阅读 6,106评论 0 4
  • 公元:2019年11月28日19时42分农历:二零一九年 十一月 初三日 戌时干支:己亥乙亥己巳甲戌当月节气:立冬...
    石放阅读 6,916评论 0 2
  • 今天上午陪老妈看病,下午健身房跑步,晚上想想今天还没有断舍离,马上做,衣架和旁边的的布衣架,一看乱乱,又想想自己是...
    影子3623253阅读 2,932评论 3 8