CANN/asc-devkit张量形状定义

Shape

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

产品支持情况

功能说明

Shape用于定义张量的形状，描述张量在各维度上的大小。

类型定义

Shape是一个模板别名，用于表示张量的形状：

template <typename... Shapes> using Shape = Std::tuple<Shapes...>;

其中：

• Shapes...是可变参数模板，表示各维度的大小
• 实际类型为Std::tuple<Shapes...>

参数说明

API映射关系

Shape通常通过MakeShape函数创建

约束说明

• Shape的维度数量不能超过硬件支持的最大维度数。
• 各维度的值必须为正整数。
• 支持的数据类型包括：size_t、int等整数类型或者Std::Int类型。

调用示例

// 使用整数类型创建一个3维张量的Shape auto shape = AscendC::Te::MakeShape(10, 20, 30); // 获取各维度的大小 auto dim0 = AscendC::Std::get<0>(shape); // dim0 = 10 auto dim1 = AscendC::Std::get<1>(shape); // dim1 = 20 auto dim2 = AscendC::Std::get<2>(shape); // dim2 = 30 // 使用Std::Int创建一个3维张量的Shape auto shapeInt = AscendC::Te::MakeShape(AscendC::Std::Int<10>{}, AscendC::Std::Int<20>{}, AscendC::Std::Int<30>{}); // 获取各维度的大小 auto dimInt0 = AscendC::Std::get<0>(shapeInt); // dimInt0 = 10 auto dimInt1 = AscendC::Std::get<1>(shapeInt); // dimInt1 = 20 auto dimInt2 = AscendC::Std::get<2>(shapeInt); // dimInt2 = 30

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言，原生支持C和C++标准规范，主要由类库和语言扩展层构成，提供多层级API，满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit