在matlab中没有for循环的行和列的矩阵乘法

时间:2017-02-13 15:15:20

标签: matlab matrix

我有以下问题,如果没有for循环,就无法想到办法。

考虑两个矩阵:

a=rand(N,3), b=rand(3,N)

我想得到的是:b(:, i)*a(i, :)(3 * 3矩阵)在第三维中连接所有i

因此,对于上面的例子,结果应该是(3 * 3 * N)矩阵。

1 个答案:

答案 0 :(得分:6)

Matlab R2016b版本:

c = reshape(a.',[1,3,N]) .* reshape(b,[3,1,N]);

早期的Matlab版本:

c = repmat(reshape(a.',[1,3,N]),[3,1,1]) .* repmat(reshape(b,[3,1,N]),[1,3,1]);

编辑:这是Matlab R2016b(Win7x64)的快速基准测试。矢量化的加速度约为50倍。

Benchmark on R2016b (Win7x64)