MATLAB Answers

Matrix multiplication of 3d arrays

164 ビュー (過去 30 日間)
Riaz Patel
Riaz Patel 2018 年 5 月 16 日
回答済み: michio 2020 年 10 月 12 日
I have a 3d array of dimension 2*5000*150 and another array of dimension 5000*1*150. I would like to take each 2*5000 matrix in the first 3d array and multiply it with the corresponding 5000*1 matrix in the second 3d array.
I need to do this in as efficient a way as possible and without any additional functions that are not preinstalled on a standard matlab installation.

回答 (3 件)

michio 2020 年 10 月 12 日

Ameer Hamza
Ameer Hamza 2018 年 5 月 16 日
Just use for loop
result = zeros(2,1,150);
for i=1:150
result(:,:,i) = matrix1(:,:,i)*matrix2(:,:,i);
  2 件のコメント
Ameer Hamza
Ameer Hamza 2018 年 5 月 16 日
The MATLAB syntax does not allow such mixing of element-wise and matrix multiplication between two matrices in a single command. Any other manipulation you will try to do will just add extra time or memory overhead. For example, you can make you code compact to just one line
result = cell2mat(cellfun(@(x,y) x*y, num2cell(x,[1 2]), num2cell(y,[1 2]), 'UniformOutput', false));
but this is very slower than for loop. MATLAB matrix operations are already very optimized so even creating a mex file is unlikely to help. for loop will be the fastest you can get in this case.


James Tursa
James Tursa 2018 年 5 月 16 日
編集済み: James Tursa 2018 年 5 月 16 日
Some options from the FEX:
MTIMESX: (C-mex code requiring C compiler) needs updated build routine for later MATLAB versions
MMX: (C-mex code requiring C compiler)
And raw times code:
result = sum(bsxfun(@times,matrix1,reshape(matrix2,1,5000,150)),2);
result = sum(matrix1.*reshape(matrix2,1,5000,150),2); % later versions of MATLAB

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by