昇腾社区首页
中文
注册

aclGetStorageShape

Atlas 200/300/500 推理产品,不支持该接口。

函数功能

获取Tensor在内存上的存储shape。

函数原型

aclnnStatus aclGetStorageShape(const aclTensor *tensor, int64_t **storageDims, uint64_t *storageDimsNum)

参数说明

参数名

输入/输出

说明

tensor

输入

输入tensor的地址。

需提前调用aclCreateTensor接口创建aclTensor。

storageDims

输出

表示存储shape的维度值。

storageDimsNum

输出

表示存储shape的维度数。

返回值说明

返回0表示成功,返回其它值表示失败。