函数:get_cur_output_dims
产品支持情况
产品 |
是否支持 |
---|---|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
功能说明
根据模型描述信息获取模型支持的动态宽高信息。
函数原型
- C函数原型
1
aclError aclmdlGetCurOutputDims(const aclmdlDesc *modelDesc, size_t index, aclIODims *dims)
- python函数
1
dims, ret = acl.mdl.get_cur_output_dims(model_desc, index)
参数说明
参数名 |
说明 |
---|---|
model_desc |
int,aclmdlDesc类型数据的指针地址。 需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。 |
index |
int,指定获取第几个输出的Dims,index值从0开始。 |
返回值说明
返回值 |
说明 |
---|---|
dims |
dict,输出实际维度信息。
|
ret |
int,错误码。
|
约束说明
当前仅支持通过本接口获取以下场景中的模型输出Tensor的维度信息:
- 通过模型转换设置多档Batch size或分辨率或维度值,实现动态Batch或动态分辨率或动态维度(ND格式)时。
- 如果用户已调用acl.mdl.set_dynamic_batch_size设置Batch或调用acl.mdl.set_dynamic_hw_size接口设置输入图片的宽高或调用acl.mdl.set_input_dynamic_dims接口设置某动态维度的值,则可通过该接口获取指定模型输出Tensor的实际维度信息。
- 如果用户未调用acl.mdl.set_dynamic_batch_size接口、acl.mdl.set_dynamic_hw_size接口或acl.mdl.set_input_dynamic_dims接口,则通过该接口可获取最大档的维度信息。
- 固定Shape场景下,通过该接口获取指定的模型输出Tensor的维度信息。
父主题: aclmdlDesc