c++ - 通过另一个 vector 推力访问 vector 元素

Question

在Thrust我有三个设备 vector

device_vector<int> ID(N)
device_vector<float> F(M),Y(N)

通常在哪里 M<<N (但这应该无关紧要)。 ID 的所有值在 0...M-1 范围内.

我现在想分配 Y[i] = F[ID[i]]对于所有 i=0...N-1 .如果我正确理解 Thrust，我不能用 for_each 做到这一点和一些仿函数。有没有什么方法可以用 thrust::transform 来实现呢？或类似的东西？

Answer 1

您可以使用 thrust:gather 执行此操作，类似于:

device_vector<int> ID(N);
device_vector<float> F(M),Y(N);

thrust::gather(ID.begin(), ID.end(), F.begin(), Y.begin());

在调用中，ID 是用于收集F 中的值并将它们写入Y 的映射。

[标准免责声明:用浏览器编写，未经测试，使用风险自负]