矢量化嵌套索引

Question

我有一个 for 循环来做索引：

for (int i=0; i<N; i++){
    a[i] = b[c[i]]
}

c 是感兴趣的索引，是 int *，而 b 和 a 是 float * 和操纵值。

但是，这需要很长时间（而且不能花那么长时间）。我想要一些矢量化版本，很可能在 BLAS/LAPLACK/etc 中找到。

我正在寻找nested_indexing(float * output_vector, float * input_vector, int * input_indices)。

我试过查看docs，但没有找到任何东西。

Answer 1

vDSP_vgathr 正是这样做的。它接受两个float * 和一个int *。它相当于for (i=0; i<N; i++) a[i] = b[c[i]]。

他们使用的措辞是

使用向量 B 的元素作为索引，将向量 A 的选定元素复制到向量 C 中的连续位置

也许它也可以是顺序索引。我注意到，要找到这些晦涩难懂的函数，最难的部分是找到要在搜索中使用的正确词。