Shape【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√功能说明Shape用于定义张量的形状描述张量在各维度上的大小。类型定义Shape是一个模板别名用于表示张量的形状template typename... Shapes using Shape Std::tupleShapes...;其中Shapes...是可变参数模板表示各维度的大小实际类型为Std::tupleShapes...参数说明参数名输入/输出描述Shapes...输入各维度的大小类型为size_t等整数类型或者Std::Int类型。API映射关系Shape通常通过MakeShape函数创建约束说明Shape的维度数量不能超过硬件支持的最大维度数。各维度的值必须为正整数。支持的数据类型包括size_t、int等整数类型或者Std::Int类型。调用示例// 使用整数类型创建一个3维张量的Shape auto shape AscendC::Te::MakeShape(10, 20, 30); // 获取各维度的大小 auto dim0 AscendC::Std::get0(shape); // dim0 10 auto dim1 AscendC::Std::get1(shape); // dim1 20 auto dim2 AscendC::Std::get2(shape); // dim2 30 // 使用Std::Int创建一个3维张量的Shape auto shapeInt AscendC::Te::MakeShape(AscendC::Std::Int10{}, AscendC::Std::Int20{}, AscendC::Std::Int30{}); // 获取各维度的大小 auto dimInt0 AscendC::Std::get0(shapeInt); // dimInt0 10 auto dimInt1 AscendC::Std::get1(shapeInt); // dimInt1 20 auto dimInt2 AscendC::Std::get2(shapeInt); // dimInt2 30【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考