推力計數迭代內部功能

Question

非常感謝你的答案在Cuda Thrust Custom function

最後一兩件事，如果我想通過另一種載體複製的值，如本？

for (i=0;i<n;i++) 
     for (j=0;j<n;j++) 
      y[i*n+j]=h1[i]*a1[pos*n+j]; 

它仍然不清楚如何將值傳遞給創建的函數謝謝！

Answer 1

thrust::transform(
    thrust::make_permutation_iterator(h1.begin(), 
     thrust::make_transform_iterator(thrust::make_counting_iterator(0), 
      IndexDivFunctor(n))), 
    thrust::make_permutation_iterator(h1.begin(), 
     thrust::make_transform_iterator(thrust::make_counting_iterator(0), 
      IndexDivFunctor(n))) + n * n, 
    thrust::make_permutation_iterator(a1.begin(), 
     thrust::make_transform_iterator(thrust::make_counting_iterator(0), 
      Indexa1Functor(n, pos))), 
    y.begin(), 
    thrust::multiplies<double>());