cuda - 使用 CUDA Thrust 进行逐元素幂运算

Question

有没有办法用 pow 来转换推力矢量？功能？换句话说，我想变换每个元素x向量到pow(x,a) ，与 a一个常数。

Answer 1

请引用Section Transformations在 Thrust 快速入门指南中了解如何编写具有初始化参数的仿函数。

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;
        }
};