c - 向量化嵌套索引

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]] .

他们使用的措辞是

Uses elements of vector B as indices to copy selected elements of vector A to sequential locations in vector C

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