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

1 view (last 30 days)
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?

Accepted Answer

Walter Roberson
Walter Roberson on 2 Apr 2016

More Answers (0)

Categories

Find more on Matrices and Arrays in Help Center and File Exchange

Community Treasure Hunt

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

Start Hunting!