Multi Dimension Matrix Elements multiplication

Question

sko 2021-2-24

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/754819-multi-dimension-matrix-elements-multiplication

回答： Steven Lord 2021-2-24

Hello,

I am strugling with some matrix multiplication...

I am having one M:

matrix and one N:

matrix with some given elements. Elements of M are given as

whereas elements of N are given as

. I wish to obtain maxtrix P:

whose elements

,

and

. Thus, matrix

would be

in my case.

Does anyone know how to perform this complex operation in the fastest way (without loops)?

Thanks in advance

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Rik 2021-2-24

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/754819-multi-dimension-matrix-elements-multiplication#answer_632054

在 MATLAB Online 中打开

Not completely without loops, but this should be fast enough. You should also be aware that loops can sometimes be the fastest way.

M=rand(3,2);N=rand(2,3,4);
%prime the execution engine to produce a more reliable comparison
timeit(@()option1(M,N));
timeit(@()option1(M,N))
ans = 1.5263e-05
timeit(@()option2(M,N))
ans = 5.3950e-06
isequal(option1(M,N),option2(M,N))
ans = logical
   1
function P=option1(M,N)
P=zeros(size(M,1),size(N,2),size(N,3));
for n=1:size(M,1)
    P(n,:,:)=M(n,1)*N(1,:,:)+M(n,2)*N(2,:,:);
end
end
function P=option2(M,N)
P=zeros(size(M,1),size(N,2),size(N,3));
for i=1:size(M,1)
    for j=1:size(N,2)
        for k=1:size(N,3)
            P(i,j,k)=M(i,1)*N(1,j,k)+M(i,2)*N(2,j,k);
        end
    end
end
end