cuda - CUDA Thrust を使用した要素単位のべき乗演算

Question

pow推力ベクトルを関数で変換する方法はありますか? xつまり、ベクトルの各要素を定数pow(x,a)でに変換したいのです。a

Answer 1

初期化されたパラメーターを使用してファンクターを作成する方法については、Thrust クイック スタート ガイドのセクション 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;
        }
};