Star0

Star0

cann-robotfix(index): 修复 median、nan_median 和 kth_value 的 C++ codecheck 及接口问题

b160221b创建于 9月17日历史提交

文件	最后提交记录	最后更新时间
docs	feat(index): migrate kth_value, median and nan_median from ops-math Co-authored-by: ConanHuang<huangxiaobin1@huawei.com> # message auto-generated for no-merge-commit merge: !10222 merge master into master feat(index): migrate kth_value, median and nan_median from ops-math Created-by: ConanHuang Commit-by: ConanHuang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 将 ops-math 的 kth_value、median、nan_median 迁入 index 对应目录，同时将 GatherV2 目录中的五个 ACLNN 接口归位，集中维护实现、接口、测试和文档。 - 承接三个算子的 Ascend950/arch35 Kernel、Host/Tiling、InferShape、L0 API、UT 和 ST 数据。主要算法、TilingKey、TilingData 保持原有设计；三个 golden.py 和三份 ST CSV 与源仓一致。 - 将依赖的 Sort 公共代码复制到 kth_value/op_host/arch35 与 op_kernel/arch35/common，调整命名和包含路径；适配目标仓 CMake、KERNEL_SRC、编译选项及公共头文件，UT 保持 arch35 隔离。分类归入 sort@ops-nn。 - 将 aclnnKthvalue、aclnnMedian、aclnnMedianDim、aclnnNanMedian、aclnnNanMedianDim 的实现、示例、文档和测试从 gather_v2 移到各自算子目录。删除三算子的 common 桩和弱符号判断，保留 IsRegbase 芯片路径选择。 - Median/NanMedian 的共用辅助函数由 `median/op_api/median_common.cpp` 单份定义，`median_common.h` 仅供内部 cpp 共享声明，通过 L0 函数指针避免 Median 反向依赖 NanMedian。这些辅助函数和常量原本属于 cpp 内部实现，未在原公共头文件中暴露；`aclnn_median.h` 不包含该内部头文件，安装规则也不将其作为公开 ACLNN 头文件导出。 - CMake 保留真实 ACLNN 调用依赖：KthValue 的旧芯片 Sort+Gather 路径仍依赖 GatherV2；Median/NanMedian 声明所需 KthValue、GatherV2、MedianDim、GatherElements 及共享实现依赖。GatherV2 移除已迁出调用方的 MedianDim 依赖。接口兼容性：公共头文件统一为 `index/median/op_api/aclnn_median.h`，保留 aclnnMedian、aclnnMedianDim、aclnnNanMedian、aclnnNanMedianDim 的全部八个两阶段接口声明，函数签名及 `extern "C"` 保持不变。调用方仍包含 `aclnnop/aclnn_median.h`，无需修改原有 include；不新增 `aclnn_nan_median.h`。Median 与 NanMedian 的 cpp 按算子目录拆分，示例、测试及文档引用同步统一。配套源仓清理 PR：https://gitcode.com/cann/ops-math/merge_requests/5550 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5783 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 5个aclnn接口各自500+ torch用例通过 - 3个算子总共500+ST用例通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 新增 kth_value、median、nan_median README，补充 op_list.md、op_api_list 及相关目录索引，修正 ACLNN 文档的源码链接与头文件引用。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [x] 其他，请描述：跨仓算子迁移及接口归位 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!10222	1 个月前
examples	feat(index): migrate kth_value, median and nan_median from ops-math Co-authored-by: ConanHuang<huangxiaobin1@huawei.com> # message auto-generated for no-merge-commit merge: !10222 merge master into master feat(index): migrate kth_value, median and nan_median from ops-math Created-by: ConanHuang Commit-by: ConanHuang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 将 ops-math 的 kth_value、median、nan_median 迁入 index 对应目录，同时将 GatherV2 目录中的五个 ACLNN 接口归位，集中维护实现、接口、测试和文档。 - 承接三个算子的 Ascend950/arch35 Kernel、Host/Tiling、InferShape、L0 API、UT 和 ST 数据。主要算法、TilingKey、TilingData 保持原有设计；三个 golden.py 和三份 ST CSV 与源仓一致。 - 将依赖的 Sort 公共代码复制到 kth_value/op_host/arch35 与 op_kernel/arch35/common，调整命名和包含路径；适配目标仓 CMake、KERNEL_SRC、编译选项及公共头文件，UT 保持 arch35 隔离。分类归入 sort@ops-nn。 - 将 aclnnKthvalue、aclnnMedian、aclnnMedianDim、aclnnNanMedian、aclnnNanMedianDim 的实现、示例、文档和测试从 gather_v2 移到各自算子目录。删除三算子的 common 桩和弱符号判断，保留 IsRegbase 芯片路径选择。 - Median/NanMedian 的共用辅助函数由 `median/op_api/median_common.cpp` 单份定义，`median_common.h` 仅供内部 cpp 共享声明，通过 L0 函数指针避免 Median 反向依赖 NanMedian。这些辅助函数和常量原本属于 cpp 内部实现，未在原公共头文件中暴露；`aclnn_median.h` 不包含该内部头文件，安装规则也不将其作为公开 ACLNN 头文件导出。 - CMake 保留真实 ACLNN 调用依赖：KthValue 的旧芯片 Sort+Gather 路径仍依赖 GatherV2；Median/NanMedian 声明所需 KthValue、GatherV2、MedianDim、GatherElements 及共享实现依赖。GatherV2 移除已迁出调用方的 MedianDim 依赖。接口兼容性：公共头文件统一为 `index/median/op_api/aclnn_median.h`，保留 aclnnMedian、aclnnMedianDim、aclnnNanMedian、aclnnNanMedianDim 的全部八个两阶段接口声明，函数签名及 `extern "C"` 保持不变。调用方仍包含 `aclnnop/aclnn_median.h`，无需修改原有 include；不新增 `aclnn_nan_median.h`。Median 与 NanMedian 的 cpp 按算子目录拆分，示例、测试及文档引用同步统一。配套源仓清理 PR：https://gitcode.com/cann/ops-math/merge_requests/5550 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5783 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 5个aclnn接口各自500+ torch用例通过 - 3个算子总共500+ST用例通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 新增 kth_value、median、nan_median README，补充 op_list.md、op_api_list 及相关目录索引，修正 ACLNN 文档的源码链接与头文件引用。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [x] 其他，请描述：跨仓算子迁移及接口归位 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!10222	1 个月前
op_api	feat(index): migrate kth_value, median and nan_median from ops-math Co-authored-by: ConanHuang<huangxiaobin1@huawei.com> # message auto-generated for no-merge-commit merge: !10222 merge master into master feat(index): migrate kth_value, median and nan_median from ops-math Created-by: ConanHuang Commit-by: ConanHuang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 将 ops-math 的 kth_value、median、nan_median 迁入 index 对应目录，同时将 GatherV2 目录中的五个 ACLNN 接口归位，集中维护实现、接口、测试和文档。 - 承接三个算子的 Ascend950/arch35 Kernel、Host/Tiling、InferShape、L0 API、UT 和 ST 数据。主要算法、TilingKey、TilingData 保持原有设计；三个 golden.py 和三份 ST CSV 与源仓一致。 - 将依赖的 Sort 公共代码复制到 kth_value/op_host/arch35 与 op_kernel/arch35/common，调整命名和包含路径；适配目标仓 CMake、KERNEL_SRC、编译选项及公共头文件，UT 保持 arch35 隔离。分类归入 sort@ops-nn。 - 将 aclnnKthvalue、aclnnMedian、aclnnMedianDim、aclnnNanMedian、aclnnNanMedianDim 的实现、示例、文档和测试从 gather_v2 移到各自算子目录。删除三算子的 common 桩和弱符号判断，保留 IsRegbase 芯片路径选择。 - Median/NanMedian 的共用辅助函数由 `median/op_api/median_common.cpp` 单份定义，`median_common.h` 仅供内部 cpp 共享声明，通过 L0 函数指针避免 Median 反向依赖 NanMedian。这些辅助函数和常量原本属于 cpp 内部实现，未在原公共头文件中暴露；`aclnn_median.h` 不包含该内部头文件，安装规则也不将其作为公开 ACLNN 头文件导出。 - CMake 保留真实 ACLNN 调用依赖：KthValue 的旧芯片 Sort+Gather 路径仍依赖 GatherV2；Median/NanMedian 声明所需 KthValue、GatherV2、MedianDim、GatherElements 及共享实现依赖。GatherV2 移除已迁出调用方的 MedianDim 依赖。接口兼容性：公共头文件统一为 `index/median/op_api/aclnn_median.h`，保留 aclnnMedian、aclnnMedianDim、aclnnNanMedian、aclnnNanMedianDim 的全部八个两阶段接口声明，函数签名及 `extern "C"` 保持不变。调用方仍包含 `aclnnop/aclnn_median.h`，无需修改原有 include；不新增 `aclnn_nan_median.h`。Median 与 NanMedian 的 cpp 按算子目录拆分，示例、测试及文档引用同步统一。配套源仓清理 PR：https://gitcode.com/cann/ops-math/merge_requests/5550 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5783 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 5个aclnn接口各自500+ torch用例通过 - 3个算子总共500+ST用例通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 新增 kth_value、median、nan_median README，补充 op_list.md、op_api_list 及相关目录索引，修正 ACLNN 文档的源码链接与头文件引用。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [x] 其他，请描述：跨仓算子迁移及接口归位 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!10222	1 个月前
op_host	fix(index): 修复 median、nan_median 和 kth_value 的 C++ codecheck 及接口问题 Co-authored-by: ConanHuang<huangxiaobin1@huawei.com> # message auto-generated for no-merge-commit merge: !10515 merge fix/median-kthvalue-codecheck into master fix(index): 修复 median、nan_median 和 kth_value 的 C++ codecheck 及接口问题 Created-by: ConanHuang Commit-by: ConanHuang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修复 median、nan_median 和 kth_value 的 C++ codecheck 及接口问题： - 为只读 InferShape/Tiling context 参数补充 `const`； - 移除头文件冗余 include 和全局 `using namespace`，补充精确命名空间限定； - 增加维度合法性校验，并使用安全的有符号/无符号转换； - 修复 nan_median 标量分支对 `ZerosLike` 返回值的错误空指针校验； - 新增 `aclnn_nan_median.h`，将 `aclnnNanMedian` 和 `aclnnNanMedianDim` 两段式接口声明迁移到 nan_median 算子目录； - 在原 `aclnn_median.h` 中转发包含新头文件，保持已有调用代码的头文件兼容性，并同步更新实现、UT、示例和接口文档。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5862 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 二级冒烟通过 - 四个aclnn接口各自抽取50个用例通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> - 更新 `aclnnNanMedian.md` 和 `aclnnNanMedianDim.md`，示例改为包含 `aclnnop/aclnn_nan_median.h`。 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!10515	1 个月前
op_kernel	kth_value,median,nan_median算子新增ascend350平台支持 Co-authored-by: jerry_gd<jerry.luo@huawei.com> # message auto-generated for no-merge-commit merge: !10327 merge master into master kth_value,median,nan_median算子新增ascend350平台支持 Created-by: jerry_gd Commit-by: jerry_gd Merged-by: cann-robot Description: ## 描述 kth_value,median,nan_median算子新增ascend350平台支持 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5838 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> bash build.sh --pkg --soc=ascend350 --ops="kth_value,median,nan_median"，编译验证通过。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> NA ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!10327	1 个月前
tests	feat(index): migrate kth_value, median and nan_median from ops-math Co-authored-by: ConanHuang<huangxiaobin1@huawei.com> # message auto-generated for no-merge-commit merge: !10222 merge master into master feat(index): migrate kth_value, median and nan_median from ops-math Created-by: ConanHuang Commit-by: ConanHuang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 将 ops-math 的 kth_value、median、nan_median 迁入 index 对应目录，同时将 GatherV2 目录中的五个 ACLNN 接口归位，集中维护实现、接口、测试和文档。 - 承接三个算子的 Ascend950/arch35 Kernel、Host/Tiling、InferShape、L0 API、UT 和 ST 数据。主要算法、TilingKey、TilingData 保持原有设计；三个 golden.py 和三份 ST CSV 与源仓一致。 - 将依赖的 Sort 公共代码复制到 kth_value/op_host/arch35 与 op_kernel/arch35/common，调整命名和包含路径；适配目标仓 CMake、KERNEL_SRC、编译选项及公共头文件，UT 保持 arch35 隔离。分类归入 sort@ops-nn。 - 将 aclnnKthvalue、aclnnMedian、aclnnMedianDim、aclnnNanMedian、aclnnNanMedianDim 的实现、示例、文档和测试从 gather_v2 移到各自算子目录。删除三算子的 common 桩和弱符号判断，保留 IsRegbase 芯片路径选择。 - Median/NanMedian 的共用辅助函数由 `median/op_api/median_common.cpp` 单份定义，`median_common.h` 仅供内部 cpp 共享声明，通过 L0 函数指针避免 Median 反向依赖 NanMedian。这些辅助函数和常量原本属于 cpp 内部实现，未在原公共头文件中暴露；`aclnn_median.h` 不包含该内部头文件，安装规则也不将其作为公开 ACLNN 头文件导出。 - CMake 保留真实 ACLNN 调用依赖：KthValue 的旧芯片 Sort+Gather 路径仍依赖 GatherV2；Median/NanMedian 声明所需 KthValue、GatherV2、MedianDim、GatherElements 及共享实现依赖。GatherV2 移除已迁出调用方的 MedianDim 依赖。接口兼容性：公共头文件统一为 `index/median/op_api/aclnn_median.h`，保留 aclnnMedian、aclnnMedianDim、aclnnNanMedian、aclnnNanMedianDim 的全部八个两阶段接口声明，函数签名及 `extern "C"` 保持不变。调用方仍包含 `aclnnop/aclnn_median.h`，无需修改原有 include；不新增 `aclnn_nan_median.h`。Median 与 NanMedian 的 cpp 按算子目录拆分，示例、测试及文档引用同步统一。配套源仓清理 PR：https://gitcode.com/cann/ops-math/merge_requests/5550 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5783 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 5个aclnn接口各自500+ torch用例通过 - 3个算子总共500+ST用例通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 新增 kth_value、median、nan_median README，补充 op_list.md、op_api_list 及相关目录索引，修正 ACLNN 文档的源码链接与头文件引用。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [x] 其他，请描述：跨仓算子迁移及接口归位 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!10222	1 个月前
CMakeLists.txt	kth_value,median,nan_median算子新增ascend350平台支持 Co-authored-by: jerry_gd<jerry.luo@huawei.com> # message auto-generated for no-merge-commit merge: !10327 merge master into master kth_value,median,nan_median算子新增ascend350平台支持 Created-by: jerry_gd Commit-by: jerry_gd Merged-by: cann-robot Description: ## 描述 kth_value,median,nan_median算子新增ascend350平台支持 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5838 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> bash build.sh --pkg --soc=ascend350 --ops="kth_value,median,nan_median"，编译验证通过。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> NA ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!10327	1 个月前
README.md	feat(index): migrate kth_value, median and nan_median from ops-math Co-authored-by: ConanHuang<huangxiaobin1@huawei.com> # message auto-generated for no-merge-commit merge: !10222 merge master into master feat(index): migrate kth_value, median and nan_median from ops-math Created-by: ConanHuang Commit-by: ConanHuang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 将 ops-math 的 kth_value、median、nan_median 迁入 index 对应目录，同时将 GatherV2 目录中的五个 ACLNN 接口归位，集中维护实现、接口、测试和文档。 - 承接三个算子的 Ascend950/arch35 Kernel、Host/Tiling、InferShape、L0 API、UT 和 ST 数据。主要算法、TilingKey、TilingData 保持原有设计；三个 golden.py 和三份 ST CSV 与源仓一致。 - 将依赖的 Sort 公共代码复制到 kth_value/op_host/arch35 与 op_kernel/arch35/common，调整命名和包含路径；适配目标仓 CMake、KERNEL_SRC、编译选项及公共头文件，UT 保持 arch35 隔离。分类归入 sort@ops-nn。 - 将 aclnnKthvalue、aclnnMedian、aclnnMedianDim、aclnnNanMedian、aclnnNanMedianDim 的实现、示例、文档和测试从 gather_v2 移到各自算子目录。删除三算子的 common 桩和弱符号判断，保留 IsRegbase 芯片路径选择。 - Median/NanMedian 的共用辅助函数由 `median/op_api/median_common.cpp` 单份定义，`median_common.h` 仅供内部 cpp 共享声明，通过 L0 函数指针避免 Median 反向依赖 NanMedian。这些辅助函数和常量原本属于 cpp 内部实现，未在原公共头文件中暴露；`aclnn_median.h` 不包含该内部头文件，安装规则也不将其作为公开 ACLNN 头文件导出。 - CMake 保留真实 ACLNN 调用依赖：KthValue 的旧芯片 Sort+Gather 路径仍依赖 GatherV2；Median/NanMedian 声明所需 KthValue、GatherV2、MedianDim、GatherElements 及共享实现依赖。GatherV2 移除已迁出调用方的 MedianDim 依赖。接口兼容性：公共头文件统一为 `index/median/op_api/aclnn_median.h`，保留 aclnnMedian、aclnnMedianDim、aclnnNanMedian、aclnnNanMedianDim 的全部八个两阶段接口声明，函数签名及 `extern "C"` 保持不变。调用方仍包含 `aclnnop/aclnn_median.h`，无需修改原有 include；不新增 `aclnn_nan_median.h`。Median 与 NanMedian 的 cpp 按算子目录拆分，示例、测试及文档引用同步统一。配套源仓清理 PR：https://gitcode.com/cann/ops-math/merge_requests/5550 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/5783 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> - 5个aclnn接口各自500+ torch用例通过 - 3个算子总共500+ST用例通过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> 新增 kth_value、median、nan_median README，补充 op_list.md、op_api_list 及相关目录索引，修正 ACLNN 文档的源码链接与头文件引用。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [x] 其他，请描述：跨仓算子迁移及接口归位 ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!10222	1 个月前

KthValue

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	×
Atlas A2 训练系列产品/Atlas A2 推理系列产品	×
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

算子功能：返回输入张量x在指定维度dim上的第k个最小值及其在原输入中的索引。
计算流程：对沿dim维度取出的每个一维切片，设升序排列后的值为s，p[j]表示s[j]在该切片排序前的位置（从0开始计数），则：
$\quad indices = p[k - 1]$
例如输入张量x=[[3, 1, 2], [6, 4, 5]]，k=2，dim=1，得到y=[[2], [5]]，indices=[[2], [2]]。其中2和5在各自原始行中的索引均为2。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待计算的输入张量。	FLOAT、FLOAT16、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64	ND
y	输出	第k个最小值。数据类型与x相同，shape与x相同但dim维度的长度为1。	FLOAT、FLOAT16、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64	ND
indices	输出	选中元素在输入张量dim维度上的索引，从0开始。shape与y相同。	INT64	ND
k	必选属性	指定第k个最小值，取值范围为[1, x.shape[dim]]。	INT64	-
dim	可选属性	指定计算的维度，取值范围为[-rank(x), rank(x) - 1]。默认值为-1，表示最后一维。	INT64	-

约束说明

输入张量x不能为空，且维度数必须大于0。

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_kthvalue	通过aclnnKthvalue接口在指定维度上计算第k个最小值及索引。