matlab - 向量化在 3-D 矩阵上运行的代码

Question

我想将返回的数据组织sph2cart为带有向量元素的矩阵，并对这个矩阵中的每个元素进行操作（向量-向量或向量-标量计算）。这是我实现这一目标的一个例子：

lightV = zeros(1, 1, 3);
lightV(1,1,1) = 0.5;
lightV(1,1,2) = 0.4;
lightV(1,1,3) = 0.7;
[Az El] = meshgrid(0:60:360, 0:15:90);
[x y z] = sph2cart(Az*pi/180, El*pi/180, 1);
refV = zeros(size(Az,1), size(Az,2), 3);
radius = zeros(size(Az,1), size(Az,2));
for i = 1:size(Az,1)
    for j = 1:size(Az,2)
       refV(i,j,1) = -x(i,j);
       refV(i,j,2) = -y(i,j);
       refV(i,j,3) = z(i,j);
       radius(i,j) = dot(refV(i,j,:), lightV(1,1,:));
    end
 end

但是这看起来有些多余，我怎样才能使它更简洁？

Answer 1

好吧，您可以像这样矢量化您的代码：

S.lightV = [0.5, 0.4, 0.7];
[Az, El] = meshgrid(0:60:360, 0:15:90);
[S.x, S.y, S.z] = sph2cart(Az * pi/180, El * pi/180, 1);
S.refV = cat(3, -x, -y, z);
S.radius = sum(bsxfun(@times, S.refV, reshape(S.lightV, 1, 1, [])), 3);

请注意cat沿第三维连接的使用，以及替换嵌套 for 循环内的点积的组合bsxfun和。sum我还将所有内容都绑定在一个 structS中。