移除device_vector中的元素

Question

thrust :: device_vector值

thrust :: device_vector keys;

初始化後，鍵包含一些等於-1的元素。我想刪除鍵中的元素和值的相同位置。

但我不知道如何處理它並行？

Answer 1

有可能有很多方法來做到這一點。一種可能的方式：

1. 使用thrust::remove_if（documentation）模板的版本，用鑰匙作爲模板，在值，其中對應的關鍵是去除-1的元素。您將需要爲謂詞測試創建一個函子。
2. 上的按鍵使用thrust::remove（documentation）取消是-1

這裏的值是一個例子：

#include <iostream> 
#include <thrust/device_vector.h> 
#include <thrust/copy.h> 
#include <thrust/remove.h> 
#include <thrust/sequence.h> 

#define N 12 
typedef thrust::device_vector<int>::iterator dintiter; 

struct is_minus_one 
{ 
    __host__ __device__ 
    bool operator()(const int x) 
    { 
    return (x == -1); 
    } 
}; 

int main(){ 

    thrust::device_vector<int> keys(N); 
    thrust::device_vector<int> values(N); 

    thrust::sequence(keys.begin(), keys.end()); 
    thrust::sequence(values.begin(), values.end()); 

    keys[3] = -1; 
    keys[9] = -1; 

    dintiter nve = thrust::remove_if(values.begin(), values.end(), keys.begin(), is_minus_one()); 
    dintiter nke = thrust::remove(keys.begin(), keys.end(), -1); 

    std::cout << "results values:" << std::endl; 
    thrust::copy(values.begin(), nve, std::ostream_iterator<int>(std::cout, " ")); 
    std::cout << std::endl << "results keys:" << std::endl; 
    thrust::copy(keys.begin(), nke, std::ostream_iterator<int>(std::cout, " ")); 
    std::cout << std::endl; 

    return 0; 
}