추력시 uint2에서 하나의 벡터로 값을 복사하는 방법은 무엇입니까?

Question

값이 vector<uint2> results (size) 인 경우 results.y 값만 vector<int> count (size)에 복사하고 싶습니다. thrust::transform 기능을 사용하여 어떻게하면됩니까?

Answer 1

당신은 함수 객체의 어떤 종류를 선언 할 필요가 (이 될 수 있습니다 UnaryFunction 또는 uint2에서 두 번째 요소를 선택합니다 BinaryFunction 당신이 NVCC에서 --expt-extended-lambda을 사용하는 경우 당신은 람다를 사용할 수 있습니다.

auto selector = [&] __device__ (const uint2& pair) { return pair.y; };

당신을 대신에 함수 객체를 사용할 수 있습니다 다음

struct Selector 
{ 
    __host__ __device__ unsigned int operator()(const uint2& pair) 
    { 
     return pair.y; 
    } 
}; 

을 그리고 thrust::transform에서 사용 :

를 스택 오버플로

thrust::transform(results.begin(), results.end(), count.begin(), selector);

또는

Selector selectorObject; 
thrust::transform(results.begin(), results.end(), count.begin(), selectorObject);