推力计数迭代内部功能

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>());