推力で1つのベクトルにuint2から値をコピーする方法は？

Question

私はvector<uint2> results (size)の値を持っています。ただresults.yの値だけをvector<int> count (size)にコピーしたいのです。 thrust::transform機能を使用してこれを行うにはどうすればよいですか？

Answer 1

あなたは（それがuint2から2番目の要素を選択するかUnaryFunctionまたは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);