C函数原型 |
aclError aclmdlGetOutputDims(const aclmdlDesc *modelDesc, size_t index, aclmdlIODims *dims) |
---|---|
Python函数 |
dims, ret = acl.mdl.get_output_dims(model_desc, index) |
函数功能 |
根据模型描述获取指定的模型输出Tensor的维度信息。
|
输入说明 |
model_desc:int,aclmdlDesc类型数据的指针地址。 需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。 index:int,指定获取第几个输入的大小,index值从0开始。 |
返回值说明 |
dims:dict,针对多batch或多种宽高的场景,按照该字典的数组返回其支持的最大档的组合,具体请参见aclmdlIODims。 dims = { "name": xxx, #tensor name "dimCount":xxx,#shape中的维度个数 "dims": [xx, xx, xx] # 维度信息 }
ret:int,错误码。
|
约束说明 |
无 |