How do I apply a function to every 2D plane of a 3D gpuArray?

조회 수: 1 (최근 30일)
I have a gpuArray, A, of dimensions (Nx,Ny,Nz). I would like to apply a function on every (x,y) slice. Essentially:
for z = 1:Nz
E[:,:,z] = f(E[:,:,z]);
end
I would also like to perform functions along the z dimension:
for x = 1:Nx
for y = 1:Ny
E[x,y,:] = g(E[x,y,:]);
end
end
What is the most computationally efficient way of handling this problem? Is this vectorizable?

채택된 답변

Walter Roberson
Walter Roberson 2016년 4월 2일

추가 답변 (0개)

카테고리

Help CenterFile Exchange에서 Matrices and Arrays에 대해 자세히 알아보기

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by