Select【免费下载链接】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 训练系列产品不支持功能说明选择Layout的shape和stride指定维度组成新的layout对象并返回。函数原型template size_t... Is, typename Shape, typename Stride __aicore__ inline constexpr auto Select(const LayoutShape, Stride layout)参数说明表1模板参数说明 | 参数名 | 类型 | 描述 | |--------|------|------| | Shape | 输入 | 组成Layout的shape的类型即元组tuple类型。 | | Stride | 输入 | 组成Layout的stride的类型即元组tuple类型。 | | Is... | size_t | 索引序列用于编译时递归选择shape和stride的子结构。 |表2参数说明 | 参数名 | 类型 | 描述 | |--------|------|------| | layout | 输入 | Layout用于描述张量的布局。 |返回值说明返回子Layout对象。约束说明索引Is...必须在有效范围内。调用示例using namespace AscendC::Te; auto shape MakeShape(10, 20, 30); auto layout MakeLayout(shape); // 选择第0和第1维度 auto subLayout Select0, 1(layout);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考