函数：get_input_dims_range

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 训练系列产品	√
Atlas 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 200/300/500 推理产品	√

根据模型描述信息获取模型的输入tensor的维度范围。

C函数原型

aclError aclmdlGetInputDimsRange(const aclmdlDesc *modelDesc, size_t index, aclmdlIODimsRange *dimsRange)

python函数

ims_range, ret = acl.mdl.get_input_dims_range(model_desc, index)

参数名	说明
model_desc	int，aclmdlDesc类型的指针。需提前调用create_desc接口创建aclmdlDesc类型的数据。
index	int，指定获取第几个输入的维度信息，index值从0开始。

参数名

说明

model_desc

int，aclmdlDesc类型的指针。

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

index

int，指定获取第几个输入的维度信息，index值从0开始。

返回值	说明
dims_range	dict，输入维度信息的字典。参考aclmdlIODimsRange。动态Shape且配置Shape范围的场景下，返回Shape范围，动态维度上下限不相同，静态维度上下限相同。例如，输入tensor的format为NCHW，dims为[1,3,10~224,10~300]，N和C维上是固定值，H和W维上是动态范围，那么通过本接口aclmdlIODimsRange.rangeCount为4，aclmdlIODimsRange.range数组中有4个元素，值分别为：{1,1}、{3,3}、{10,224}、{10,300}。动态Shape且配置分档的场景下，不支持使用本接口，aclmdlIODimsRange.rangeCount返回0。静态Shape场景下，返回Shape范围，但各维度上下限相同。
ret	int，返回0表示成功，返回非0表示失败。

返回值

说明

dims_range

dict，输入维度信息的字典。参考aclmdlIODimsRange。

动态Shape且配置Shape范围的场景下，返回Shape范围，动态维度上下限不相同，静态维度上下限相同。
例如，输入tensor的format为NCHW，dims为[1,3,10~224,10~300]，N和C维上是固定值，H和W维上是动态范围，那么通过本接口aclmdlIODimsRange.rangeCount为4，aclmdlIODimsRange.range数组中有4个元素，值分别为：{1,1}、{3,3}、{10,224}、{10,300}。
动态Shape且配置分档的场景下，不支持使用本接口，aclmdlIODimsRange.rangeCount返回0。
静态Shape场景下，返回Shape范围，但各维度上下限相同。

ret

int，返回0表示成功，返回非0表示失败。

父主题： aclmdlDesc