在 Matlab 中优化生成随机数

Question

我有一个两个 1 x 10^3 矩阵，它的元素在模拟循环的每个步骤中都会发生变化。要在每个 for 循环步骤中找到一个新元素，我需要 2*1000 高斯随机数。循环步数等于 10^5。在这种情况下，哪种方法更省时？

1. 在 for 循环之前创建两个 10^5 x 1000 随机矩阵并在循环内调用它们。

2. 在循环的每个步骤中创建两个 1 x 10^3 随机矩阵？

Answer 1

这里有一个小代码来回答你的问题：

function rnd_time_test
rnd_test = @(n) randn(2,n,1000);
N = 1e5;
one_time = timeit(@() rnd_test(N))
loop_time = timeit(@() rnd_test_loop(N))

    function R = rnd_test_loop(n)
        for k = 1:n
            R = randn(2,1000);
        end
    end
end

结果：

one_time =
      2.245
loop_time =
      1.6311

所以，在我的机器上，使用循环会更快。