aclGetStorageShape
函数功能
获取aclTensor的StorageShape。
StorageShape表示aclTensor的实际物理排布shape,是Tensor在内存上实际存在的大小。假设StorageShape为[10, 20],表示该aclTensor在内存上是按照[10, 20]排布的。
函数原型
aclnnStatus aclGetStorageShape(const aclTensor *tensor, int64_t **storageDims, uint64_t *storageDimsNum)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
tensor |
输入 |
输入tensor的地址。需提前调用aclCreateTensor接口创建aclTensor。 |
storageDims |
输出 |
表示StorageShape的维度值。 |
storageDimsNum |
输出 |
表示StorageShape的维度数。 |
返回值说明
返回0表示成功,返回其他值表示失败。
父主题: 基础接口