请教如何在使用matlab GPU时记录arrayfun中的循环迭代变量？

最新推荐文章于 2024-01-29 23:02:00 发布

shan_yin

最新推荐文章于 2024-01-29 23:02:00 发布

阅读量705

点赞数

分类专栏：编程文章标签： Matlab GPU Arrayfun

本文链接：https://blog.csdn.net/weixin_44120272/article/details/85034976

版权

在使用Matlab 2017b的GPU功能时，遇到一个问题：在arrayfun中尝试记录循环变量z，但出现‘Array indexing is not supported’的错误。寻求解决方案，以便能将arrayfun内部的循环变量存入矩阵，进行后续处理。

摘要由CSDN通过智能技术生成

在下述代码中，我希望能记录arrayfun函数里面的循环变量z到一个矩阵中，以作进一步的操作。Matlab版本为2017b，代码运行报错为“Array indexing is not supported”。请各位大侠赐教如何才能解决这一问题。

clear all
clc
maxIterations=100;
gridSize=300;
xlim=[-0.75, -0.73];
ylim=[ 0.12,  0.14];
t=tic();
x=gpuArray.linspace(xlim(1), xlim(2), gridSize);
y=gpuArray.linspace(ylim(1), ylim(2), gridSize);
[xGrid,yGrid]=meshgrid(x,y);
Pos=gpuArray.zeros(maxIterations,1);
count=parent_fun(xGrid,yGrid,maxIterations,Pos);
count=gather(count);
gpuArrayfunTime=toc(t)
figure(1)
imagesc(x,y,count)
reset(gpuDevice(1))

%下面是用到的子函数。使用nested function的形式是希望能进行索引，但无效。
function result=parent_fun(xGrid,yGrid,maxIterations,Pos)
             function count=tar_fun(x0,y0)
                           z0=complex(x0,y0);
                           z=z0;
                           count=1;
                           while (count<=maxIterations) && (abs(z)<=2)