我如何矢量化它？

Question

new = zeros(TR);
for i1 = 1 : TR
    for i2 = 1 : TR
        for j1 = 1 : TR
            for j2 = 1 : TR
                  new(i1, i2) = new(i1, i2) + p1(i1, j1) * p2(i2, j2) * ...
                                self.radon(ind - 1, i1, i2, j1, j2) * ...
                                current(j1, j2);
            end
        end
   end
end

这里“新”的大小，p1，p2，电流是TRxTR，氡气是N_timesteps x TR x TR x TR x TR。

我在这里所做的基本上是使用转移概率 p1 * p2 * radon 为固定时间步“ind”计算上一个时间步“ind -1”的期望。

谢谢！

更新：

我能够按如下方式对其进行半矢量化：

for i1 = 1 : TR
    for i2 = 1 : TR
        new(i1, i2) = sum(sum((p1(i1, :)' * p2(i2, :)) .* ...           
                      squeeze(self.radon(ind - 1, i1, i2, :, :)) .* current));
    end
end

所以现在它在 16 秒内运行，而不是 2 分钟。任何人都可以提出更多改进建议吗？谢谢！

Answer 1

我将“如何向量化”解释为“如何使其更快”，至少回答这个问题。除非您使用 matlab 2015b 或更高版本，否则访问类属性很慢。性能测量are available here。

您可以使用临时变量来提高性能：

radonslice=self.radon(ind - 1, :, :, :, :)

然后在你的代码中使用radonslice(1, i1, i2, j1, j2)。