Calculate lsqr of two arrays on GPU

Question

Mantas Vaitonis 2018-7-19

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/411072-calculate-lsqr-of-two-arrays-on-gpu

回答： Joss Knight 2018-7-21

Hello,

Is there way to calculate lsqr of two arrays, I mean for each columns of arrays respectively. If a (NxM) and b(NxM) are gpuArrays then:

c=lsqr(a(:,1),b(:,1));

This works with no problem. Is there a way to do it on whole arrays? I know that it could be done with for loops, but it becomes bottleneck in my code.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Joss Knight 2018-7-21

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/411072-calculate-lsqr-of-two-arrays-on-gpu#answer_329772

在 MATLAB Online 中打开

The least squares solution to a system of linear equations of one variable is pretty simple to work out, so you shouldn't really need any fancy linear algebra. I haven't actually checked this in MATLAB, but by my reckoning all you need is

c = sum(a .* b) ./ sum(a .* a);