矩陣最大在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); });

如果這不正確，或者有更好的解決方案，我歡迎您提出其他建議。