第一个算子是AddcMul,经过测试,发现该算子不能正确的实现广播操作,在多个维度上进行广播时无法成功,参考此文章实现的广播操作可以正确的完成。

Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐