C函数原型 |
aclError aclmdlGetCurOutputDims(const aclmdlDesc *modelDesc, size_t index, aclIODims *dims) |
---|---|
Python函数 |
dims, ret = acl.mdl.get_cur_output_dims(model_desc, index) |
函数功能 |
根据模型描述信息获取指定的模型输出Tensor的实际维度信息。 |
输入说明 |
model_desc:int,aclmdlDesc类型数据的指针地址。 需提前调用acl.mdl.create_desc接口创建aclmdlDesc类型的数据。 index:int,指定获取第几个输出的Dims,index值从0开始。 |
返回值说明 |
dims:dict,输出实际维度信息。
ret:int,错误码。
|
约束说明 |
当前仅支持通过本接口获取以下场景中的模型输出Tensor的维度信息:
|