代码之家  ›  专栏  ›  技术社区  ›  read Read

Cuda编程中矩阵的元素指数

  •  0
  • read Read  · 技术社区  · 10 年前

    如何在Cuda编程中对矩阵进行元素指数化?

    例如:

    A=[1 3 4;6 5 2];

    我想计算:

    B=[exp(1),exp(3),exp(4);exp(6);exp(5);(2)]

    是否有一种有效且到位的方法(即B代替a)? 立方体似乎不提供矩阵上的元素操作。

    1 回复  |  直到 10 年前
        1
  •  2
  •   Community kfsone    7 年前

    我不知道是否存在对矩阵执行元素操作的库,但您可以很容易地设置CUDA内核来完成这项工作。例如,您可以给出 A 矩阵到每个线程,他们可以执行指数运算并将答案写入 B 。然后像往常一样调用CUDA内核。看一看 at this 了解如何实现内核以及如何调用它(而不是像在 gpuMM 你会做一个指数)。

    编辑 :看起来您可以使用 Thrust 和宏集 Newton ,如图所示 SO question .