下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

函数:get_input_size_by_index

C函数原型

size_t aclmdlGetInputSizeByIndex(aclmdlDesc *modelDesc, size_t index)

Python函数

size = acl.mdl.get_input_size_by_index(model_desc, index)

函数功能

根据aclmdlDesc类型的数据,获取指定输入的大小,单位为Byte。

输入说明

model_desc:int,aclmdlDesc类型数据的指针地址。

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

index:int,指定获取第几个输入的大小,index值从0开始。

返回值说明

size:int,返回指定输入的大小,单位是Byte。

  • 针对动态Batch、动态分辨率(宽高)的场景,返回最大档位对应的输入的大小。
  • 静态场景下,返回指定输入的大小。单位是Byte。

约束说明

如果模型输入的Shape是动态的、且维度的取值为-1(表示此维度可以使用大于或等于1的任意取值),则通过本接口获取的大小为0,用户需根据实际数据占用的内存大小来申请内存。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词