Fastest way to differentiate only the sides of large matricies

2 次查看（过去 30 天）

Nathan Zechar 2021-10-2

1
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1465504-fastest-way-to-differentiate-only-the-sides-of-large-matricies

评论： Nathan Zechar 2021-10-4

I believe I have found one of the fastest ways to differentiate different portions of a large matrix using built in Matlab functions. However, there is a problem to my method, there is a remaining middle value between the side values. Allow me to demonstate with some simple code which makes use of gpuArray. For my case, I need the data on a gpu.

N = 300; % Number of points in x,y,z dimensions 
S = 10;  % Number of points from side of a dimension
for i = 1:20
B  = gpuArray(rand(N,N,N));      % Load some random data
An = diff(B(1:S+1,:,:),1,1);     % Differentation from left side
Ap = diff(B(end-S:end,:,:),1,1); % Differentation from right side
A1 = B([2:S+1,N-S+1:N],:,:)-B([1:S,N-S:N-1],:,:);
% The above is the finite difference which is applied to 
% both sides of the matrix at the same time. This method
% however, is slower than using built in diff
A2 = diff(B([1:S+1,N-S:N],:,:),1,1);
% The above method is the fastest method, however there is one
% additional data point between S+1 and N - S. Deleting this
% point would add extra computational time
end

However diff is coded on the backend of Matlab, and would need to be modified for my application. If this is true, how would I go about learning how to code up my own diff function for this special case?

Thank you

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

请先登录，再进行评论。

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

采纳的回答

Matt J 2021-10-3

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1465504-fastest-way-to-differentiate-only-the-sides-of-large-matricies#answer_800364

编辑：Matt J 2021-10-3

This seems to be about as fast, at least on the GTX 1080 Ti.

T=reshape(B([1:S+1,N-S:N],:) ,S+1,2*N^2 );
A3 = reshape( diff(T,1,1), 2*S,N,N); 

5 个评论
显示 3更早的评论隐藏 3更早的评论

Matt J 2021-10-4

编辑：Matt J 2021-10-4

No, you don't need to permute.

% along 1st axis
T  = reshape(B([1:S+1,N-S:N],:) ,S+1,2,N,N );
A1 = reshape(diff(T,1,1), 2*S,N,N);
% along 2nd axis
T=reshape(B(:,[1:S+1,N-S:N],:),N,S+1,2,N);
A2=reshape( diff(T,1,2)  , N,2*S,N);
% along 3rd axis
T=reshape(B(:,:,[1:S+1,N-S:N]),N,N,S+1,2);
A3=reshape( diff(T,1,3)  , N,N,2*S);