CANN/GE动态维度获取API

aclmdlGetInputDynamicDims

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge

产品支持情况

功能说明

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

函数原型

aclError aclmdlGetInputDynamicDims(const aclmdlDesc *modelDesc, size_t index, aclmdlIODims *dims, size_t gearCount)

参数说明

aclmdlIODims dims[gearCount]; aclmdlGetInputDynamicDims(model.modelDesc, -1, dims, gearCount);

返回值说明

返回0表示成功，返回其他值表示失败，请参见aclError。

约束说明

只有在模型构建时设置了动态维度的分档信息后，才可以调用该接口获取动态维度信息。

例如，模型有三个输入，分别为data(1, 1, 40, -1)，label(1, -1)，mask(-1, -1) ， 其中-1表示动态可变。在模型转换时，dynamic_dims参数的配置示例为：--dynamic_dims="20,20,1,1;40,40,2,2;80,60,4,4"，则通过本接口获取的动态维度信息为（**aclmdlIODims**结构体内的name暂不使用）：

• 第0档：

  • **aclmdlIODims**结构体内dimCount：8，表示所有输入tensor的维度数量之和
  • **aclmdlIODims**结构体内的dims：“1,1,40,20,1,20,1,1”，表示data(1,1,40,20)+label(1,20)+mask(1,1)

• 第1档：

  • **aclmdlIODims**结构体内dimCount：8，表示所有输入tensor的维度数量之和
  • **aclmdlIODims**结构体内的dims：“1,1,40,40,1,40,2,2”，表示data(1,1,40,40)+label(1,40)+mask(2,2)

• 第2档：

  • **aclmdlIODims**结构体内dimCount：8，表示所有输入tensor的维度数量之和
  • **aclmdlIODims**结构体内的dims：“1,1,40,80,1,60,4,4”，表示data(1,1,40,80)+label(1,60)+mask(4,4)

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge