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的维度数。 |
约束与限制
参数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; |
父主题: 基础接口