矩阵最大在Matlab上的Matlab

Question

我正在从Matlab移植一些代码以在Nvidia GPU上运行。 我想不通以下方法：

B = max(A, 0)

其中A和B是矩阵。 换句话说，我需要将矩阵中的负值替换为零。 我知道如何编写与之相关的内核函数，但如果可能的话，我想坚持使用cuBLAS或岩浆调用（以避免在构建过程中添加nvcc）。

Answer 1

我想出了一些推力的方法：

thrust::transform(A, A + m*n, [](double x) { thrust::max(x,0.0); });

如果这不正确，或者有更好的解决方案，我欢迎您提出其他建议。