GetShape【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况Ascend 950PR/Ascend 950DT支持Atlas A3 训练系列产品/Atlas A3 推理系列产品不支持Atlas A2 训练系列产品/Atlas A2 推理系列产品不支持Atlas 200I/500 A2 推理产品不支持Atlas 推理系列产品AI Core不支持Atlas 推理系列产品Vector Core不支持Atlas 训练系列产品不支持功能说明需要包含的头文件为#include tensor_api/tensor.h。GetShape用于从Layout中提取Shape部分不指定模板参数Is...返回完整的Shape元组。指定模板参数Is...从Shape顶层tuple中选择对应索引的元素。指定单个索引时返回该元素指定多个索引时返回由这些顶层元素组成的新tuple。函数原型template size_t... Is, typename ShapeType, typename StrideType __aicore__ inline constexpr auto GetShape(const LayoutShapeType, StrideType layout) template size_t... Is, typename ShapeType, typename StrideType __aicore__ inline constexpr auto GetShape(LayoutShapeType, StrideType layout)参数说明表1模板参数说明参数名输入/输出描述Is...输入可选。用于选择Shape顶层tuple中的一个或多个索引。ShapeType输入Layout的shape类型。StrideType输入Layout的stride类型。表2参数说明参数名输入/输出描述layout输入Layout对象。返回值说明返回描述张量形状的Shape对象或其子结构。约束说明Is...必须在Shape顶层tuple的有效范围内。调用示例using namespace AscendC::Te; auto layout MakeLayout(MakeShape(10, 20, 30)); // 不指定Is... → 返回完整shape元组 auto fullShape GetShape(layout); // (10, 20, 30) // 指定单个索引 → 返回对应元素 auto s0 GetShape0(layout); // 10 // 指定多个索引 → 组成新tuple auto s01 GetShape0, 1(layout); // (10, 20)【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考