| 优化top_k_v2算子支持输出索引类型为int64
Co-authored-by: caoyan_huawei<caoyan11@huawei.com>
# message auto-generated for no-merge-commit merge:
!1557 merge cy_branch_2026_03_09 into master
优化top_k_v2算子支持输出索引类型为int64
Created-by: caoyan_huawei
Commit-by: caoyan_huawei
Merged-by: cann-robot
Description: ## 描述
优化top_k_v2算子形状推断问题,本次PR为该算子的形状推断函数增加了详细的调试日志输出功能,主要优化如下:
1. 当输入张量x和index的索引不匹配时,原有日志信息不足,开发人员难以快速定位具体原因。
2. 根据实际应用场景,放开top_k_v2算子中的k和indices的绑定。
## 测试
UT测试通过,整网用例通过
## 文档更新
<!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。-->
## 类型标签
<!-- [x] 表示选中 -->
- [ ] Bug修复
- [x] 新特性
- [ ] 性能优化
- [ ] 文档更新
- [ ] 其他,请描述:
See merge request: cann/ops-math!1557 | 2 个月前 |