下载
中文
注册

GetDynamicGearInfo

函数功能

获得模型支持的动态输入档位信息,支持动态Batch、动态分辨率和分档动态维度模型。

函数原型

std::vector<std::vector<uint64_t>> Model::GetDynamicGearInfo();

返回参数说明

数据结构

说明

std::vector<std::vector<uint64_t>>

模型支持的输入Tensor档位信息。

  • std::vector<std::vector<uint64_t>>为所有档位信息。
  • std::vector<uint64_t>为某一档的具体值。

返回数据结构为空,表示获取分档信息失败或者模型不属于函数功能中描述的三种模型之一,具体的原因可查看报错信息确认。

例如,模型为动态分辨率模型,std::vector<uint64_t>为某一档的高、宽值。