CANN/opbase aclGetViewShape函数文档

aclGetViewShape【免费下载链接】opbase本项目是CANN算子库的基础框架库为算子提供公共依赖文件和基础调度能力。项目地址: https://gitcode.com/cann/opbase功能说明获取aclTensor的ViewShapeaclTensor由aclCreateTensor接口创建。ViewShape表示aclTensor的逻辑shape是Tensor在实际使用中需要用到的大小。假设ViewShape为[2, 5, 20]在算子使用时表示该aclTensor可被视为一块[2, 5, 20]的数据使用。函数原型aclnnStatus aclGetViewShape(const aclTensor *tensor, int64_t **viewDims, uint64_t *viewDimsNum)参数说明参数名输入/输出说明tensor输入输入的aclTensor。需提前调用aclCreateTensor接口创建aclTensor。viewDims输出返回的viewShape的维度值。viewDimsNum输出返回的viewShape的维度数。返回值说明返回0表示成功返回其他值表示失败返回码列表参见公共接口返回码。可能失败的原因返回161001参数tensor或viewDims或viewDimsNum空指针。约束说明参数viewDims内存是本接口内部申请使用完后必须delete手动释放。调用示例假设已有aclTensor对象xTensor获取其数据类型、数据排布格式、维度、步长、偏移等属性再根据这些属性创建一个新的aclTensor对象yTensor 。关键代码示例如下仅供参考不支持直接拷贝运行。// 1.创建xTensor int64_t xViewDims {2, 4}; int64_t xStridesValue {4, 1}; // 第1维步长4第2维步长1 int64_t xStorageDims {2, 4}; xTensor aclCreateTensor(xViewDims, 2, ACL_FLOAT16, xStridesValue, 0, ACL_FORMAT_ND, xStorageDims, 2, nullptr); // 2. 获取xTensor的各种属性值 // 获取xTensor的逻辑shapeviewDims为{2, 4}, viewDimsNum为2 int64_t *viewDims nullptr; uint64_t viewDimsNum 0; auto ret aclGetViewShape(xTensor, viewDims, viewDimsNum); // 获取xTensor的数据类型为ACL_FLOAT16 aclDataType dataType aclDataType::ACL_DT_UNDEFINED; ret aclGetDataType(xTensor, dataType); // 获取xTensor的步长信息stridesValue为{4, 1}, stridesNum为2 int64_t *stridesValue nullptr; uint64_t stridesNum 0; ret aclGetViewStrides(xTensor, stridesValue, stridesNum); // 获取xTensor的首元素对于storage的偏移值offset为0 int64_t offset 0; ret aclGetViewOffset(xTensor, offset); // 获取xTensor的数据排布格式为ACL_FORMAT_ND aclFormat format aclFormat::ACL_FORMAT_UNDEFINED; ret aclGetFormat(xTensor, format); // 获取xTensor的实际物理排布shapestorageDims为{2, 4}, storageDimsNum为2 int64_t *storageDims nullptr; uint64_t storageDimsNum 0; ret aclGetStorageShape(xTensor, storageDims, storageDimsNum); // device侧地址 void *deviceAddr; // 3.根据xTensor的属性创建新的tensor aclTensor *yTensor aclCreateTensor(viewDims, viewDimsNum, dataType, stridesValue, offset, format, storageDims, storageDimsNum, deviceAddr); // 4.手动释放内存 delete[] viewDims; delete[] stridesValue; delete[] storageDims;【免费下载链接】opbase本项目是CANN算子库的基础框架库为算子提供公共依赖文件和基础调度能力。项目地址: https://gitcode.com/cann/opbase创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考