aclGetStorageShape

函数功能

获取aclTensor的StorageShape。

StorageShape表示aclTensor的实际物理排布shape，是Tensor在内存上实际存在的大小。假设StorageShape为[10, 20]，表示该aclTensor在内存上是按照[10, 20]排布的。

函数原型

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

参数说明

返回值说明

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

可能失败的原因：

返回161001：参数tensor或storageDims或storageDimsNum为空指针。

约束与限制

参数storageDims内存为本接口申请，使用完需手动释放。

调用示例

关键代码示例如下，不可以直接拷贝运行，仅供参考。

1
2
3
4
5
6
7

int64_t* storageDims = nullptr;
uint64_t storageDimsNum = 0;
// tensor先已创建好
auto ret = aclGetStorageShape(tensor, &storageDims, &storageDimsNum);
...
// 手动释放storageDims内存
delete storageDims;