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

1 次查看（过去 30 天）
Matthew Suttinger
回答： Walter Roberson 2016-4-2
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 Center 和 File 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