DeInterleave【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品 / Atlas A3 推理系列产品xAtlas A2 训练系列产品 / Atlas A2 推理系列产品xAtlas 200I/500 A2 推理产品xAtlas 推理系列产品 AI CorexAtlas 推理系列产品 Vector CorexAtlas 训练系列产品x功能说明给定源操作数src0和src1将src0和src1中的元素解交织存入结果操作数dst0和dst1中。解交织排列方式如下图所示其中每个方格代表一个元素。函数原型两个输入template typename T __aicore__ inline void DeInterleave(const LocalTensorT dst0, const LocalTensorT dst1, const LocalTensorT src0, const LocalTensorT src1, const int32_t count)一个输入template typename T __aicore__ inline void DeInterleave(const LocalTensorT dst0, const LocalTensorT dst1, const LocalTensorT src, const int32_t srcCount)参数说明表 1模板参数说明参数名描述T操作数数据类型。表 2参数说明参数名输入/输出描述dst0/dst1输出目的操作数。类型为LocalTensor支持的TPosition为VECIN/VECCALC/VECOUT。LocalTensor的起始地址需要32字节对齐。Ascend 950PR/Ascend 950DT支持的数据类型为uint8_t/int8_t/uint16_t/int16_t/half/bfloat16_t/uint32_t/int32_t/float/uint64_t/int64_tsrc/src0/src1输入源操作数。类型为LocalTensor支持的TPosition为VECIN/VECCALC/VECOUT。LocalTensor的起始地址需要32字节对齐。源操作数的数据类型需要与目的操作数保持一致。Ascend 950PR/Ascend 950DT支持的数据类型为uint8_t/int8_t/uint16_t/int16_t/half/bfloat16_t/uint32_t/int32_t/float/uint64_t/int64_tcount输入输入/输出数据元素个数dst0/dst1/src0/src1长度大小为count。count必须为偶数。srcCount输入输入数据元素个数两个输出的大小都为输入的一半。srcCount必须为偶数。返回值说明无约束说明无调用示例本样例中只展示Compute流程中的部分代码。两个输入AscendC::DeInterleave(dst0Local, dst1Local, src0Local, src1Local, 512);结果示例如下输入数据src0Local: [1 2 3 ... 512] 输入数据src1Local: [513 514 515 ... 1024] 输出数据dst0Local: [1 3 5 ... 1023] 输出数据dst1Local: [2 4 6 ... 1024]一个输入AscendC::DeInterleave(dst0Local, dst1Local, srcLocal, 512);结果示例如下输入数据srcLocal: [1 2 3 ... 512] 输出数据dst0Local: [1 3 5 ... 511] 输出数据dst1Local: [2 4 6 ... 512]【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考