昇腾社区首页
中文
注册

函数:get_input_dynamic_gear_count

产品支持情况

产品

是否支持

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

Atlas A2 训练系列产品 / Atlas 800I A2 推理产品 /A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

根据模型描述信息获取模型的输入所支持的动态维度档位数。

函数原型

  • C函数原型
    1
    aclError aclmdlGetInputDynamicGearCount(const aclmdlDesc *modelDesc, size_t index, size_t *gearCount)
    
  • python函数
    1
    gear_count, ret = acl.mdl.get_input_dynamic_gear_count(model_desc, index)
    

参数说明

参数名

说明

model_desc

int,aclmdlDesc类型模型描述信息数据的指针地址。

需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。

index

int,预留参数,当前未使用,固定设置为-1。

返回值说明

返回值

说明

gear_count

int,动态维度档位数。

例如,模型的输入Tensor是4维,在模型转换时通过“--dynamic_dims”参数设置的分档为“1, 3, 224, 224; 2, 3, 224, 224; 1, 3, 256, 256”,那么通过该接口获取的动态维度档位数为3。

ret

int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

如果模型构建时没有设置动态维度的分档,那么通过该接口获取的动态维度档位数为0。模型构建的详细说明请参见模型构建