推力：アクティブな配列要素のインデックスを返す方法

Question

推力を使用してアクティブな配列要素のインデックスを返す、つまり配列要素が1に等しいインデックスのベクトルを返すにはどうすればよいですか？

これを拡張すると、配列の寸法が与えられた多次元インデックスの場合、どのように機能しますか？

編集：現在の機能はcopy_ifでcounting_iteratorを組み合わせてエラーに

Error 15 error : no instance of overloaded function "thrust::copy_if" matches the argument list 

Answer 1

を与えている。この

template<class VoxelType> 
void VoxelVolumeT<VoxelType>::cudaThrustReduce(VoxelType *cuda_voxels) 
{ 
    device_ptr<VoxelType> cuda_voxels_ptr(cuda_voxels); 

    int active_voxel_count = thrust::count(cuda_voxels_ptr, cuda_voxels_ptr + dim.x*dim.y*dim.z, 1); 

    device_vector<VoxelType> active_voxels; 

    thrust::copy_if(make_counting_iterator(0), 
        make_counting_iterator(dim.x*dim.y*dim.z), 
        cuda_voxels_ptr, 
        active_voxels.begin(), 
        _1 == 1); 
} 

次のようになります。

#include <thrust/copy.h> 
#include <thrust/iterator/counting_iterator.h> 
#include <thrust/functional.h> 
... 
using namespace thrust; 
using namespace thrust::placeholders; 

copy_if(make_counting_iterator<int>(0), 
     make_counting_iterator<int>(array.size()), // indices from 0 to N 
     array.begin(),        // array data 
     active_indices.begin(),     // result will be written here 
     _1 == 1);         // return when an element or array is equal to 1