BlockReduceMin,块级别的规约如何获取对应的索引
收藏回复举报
BlockReduceMin,块级别的规约如何获取对应的索引
t('forum.solved') 已解决
新人帖
发表于2024-08-17 15:11:18
0 查看
我在使用Ascend C开发单算子Demo,BlockReduceMin这个API用于获取每个datablock中的最小值和对应block的索引,我的数据是Float类型8*8*8的长度,每8个数据正好是一个datablock,现在我希望获取每个block的最小值和对应的索引,但是API只提供了获取最小值,有没有什么合适快速的方法同时获取其索引

我要发帖子