CANN/Ascend C GetStride函数

GetStride【免费下载链接】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。GetStride用于从Layout中提取Stride部分不指定模板参数Is...返回完整的Stride元组。指定模板参数Is...从Stride顶层tuple中选择对应索引的元素。指定单个索引时返回该元素指定多个索引时返回由这些顶层元素组成的新tuple。函数原型template size_t... Is, typename ShapeType, typename StrideType __aicore__ inline constexpr auto GetStride(const LayoutShapeType, StrideType layout) template size_t... Is, typename ShapeType, typename StrideType __aicore__ inline constexpr auto GetStride(LayoutShapeType, StrideType layout)参数说明表1模板参数说明参数名输入/输出描述Is...输入可选。用于选择Stride顶层tuple中的一个或多个索引。ShapeType输入Layout的shape类型。StrideType输入Layout的stride类型。表2参数说明参数名输入/输出描述layout输入Layout对象。返回值说明返回描述内存访问步长的Stride对象或其子结构。约束说明Is...必须在Stride顶层tuple的有效范围内。调用示例using namespace AscendC::Te; auto layout MakeLayout(MakeShape(10, 20, 30), MakeStride(1, 10, 200)); // 不指定Is... → 返回完整stride元组 auto fullStride GetStride(layout); // (1, 10, 200) // 指定单个索引 → 返回对应元素 auto s0 GetStride0(layout); // 1 // 指定多个索引 → 组成新tuple auto s01 GetStride0, 1(layout); // (1, 10)【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考