CANN算子:利用迭代器高效实现Tensor数据切割分块处理

摘要:本文以Diagonal算子为例,介绍并详细讲解如何利用迭代器对n维Tensor进行基于位置坐标的大批量数据读取工作。 本文分享自华为云社区《CANN算子:利用迭代器高效实现Tensor数据切割分块处理》,作者: CatherineWang 。 任务场景及目标 在CANN aicpu算子开发实现

你可能感兴趣的:(CANN算子:利用迭代器高效实现Tensor数据切割分块处理)