要实现索引取值类操作(比如gather)应该用什么Ascend C api?
收藏回复举报
要实现索引取值类操作(比如gather)应该用什么Ascend C api?
t('forum.solved') 已解决
新人帖
发表于2023-11-21 02:42:06
0 查看

在CANN文档的Ascend C的算子开发api里翻了一遍,好像并没有看到什么接口能实现向量指令级别的索引取值。那么要在AI core上去实现gather,即output[i]=input[index[i]]这种操作,应该如何有效地去实现呢?还是说AI core就是不适合干这种事,因此是AI cpu去实现的?

我要发帖子