使用CUDA Thrust的元素功率運算

Question

有沒有用pow函數轉換推力矢量的方法？換句話說，我想將矢量的每個元素x變換爲pow(x,a)，其中a是常數。

Answer 1

請參考推力Quict入門指南中的Section Transformations，瞭解如何使用初始化參數編寫仿函數。

struct saxpy_functor 
{ 
    const float a; 

    saxpy_functor(float _a) : a(_a) {} 

    __host__ __device__ 
     float operator()(const float& x, const float& y) const { 
      return a * x + y; 
     } 
}; 

Answer 2

這是一個完整的例子。正如@Eric所提到的，所有需要的是定義您自己的功能仿函數，並使用thrust::transform。

#include <thrust/sequence.h> 
#include <thrust/device_vector.h> 

class power_functor { 

    double a; 

    public: 

     power_functor(double a_) { a = a_; } 

     __host__ __device__ double operator()(double x) const 
     { 
      return pow(x,a); 
     } 
}; 

void main() { 

    int N = 20; 

    thrust::device_vector<double> d_n(N); 
    thrust::sequence(d_n.begin(), d_n.end()); 

    thrust::transform(d_n.begin(),d_n.end(),d_n.begin(),power_functor(2.)); 

    for (int i=0; i<N; i++) { 
     double val = d_n[i]; 
     printf("Device vector element number %i equal to %f\n",i,val); 
    } 

    getchar(); 
}