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

2016 4 2

1 个回答

3 次查看（30 天）

0 个投票

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?

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

Walter Roberson 2016-4-2

2 个投票

在帮助中心和 File Exchange 中查找有关 Matrix Indexing 的更多信息

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

Translated by