下载
EN
注册

aclGetViewShape

函数功能

获取aclTensor的ViewShape。

ViewShape表示aclTensor的逻辑shape,是Tensor在实际使用中需要用到的大小。假设ViewShape为[2, 5, 20],在算子使用时,表示该aclTensor可被视为一块[2, 5, 20]的数据使用。

函数原型

aclnnStatus aclGetViewShape(const aclTensor *tensor, int64_t **viewDims, uint64_t *viewDimsNum)

参数说明

参数名

输入/输出

说明

tensor

输入

输入的tensor。需提前调用aclCreateTensor接口创建aclTensor。

viewDims

输出

返回的viewShape的维度值。

viewDimsNum

输出

返回的viewShape的维度数。

返回值说明

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

可能失败的原因:

返回161001:参数tensor或viewDims或viewDimsNum为空指针。

约束与限制

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

调用示例

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

1
2
3
4
5
6
7
int64_t* viewDims = nullptr;
uint64_t viewDimsNum = 0;
// tensor先已创建好
auto ret = aclGetViewShape(tensor, &viewDims, &viewDimsNum);
...
// 手动释放viewDims内存
delete viewDims;