How to efficiently calculate a weighted sum of 3 dimensional matrix through a vector in matlab

5 views (last 30 days)
A is a 3 dimensional matrix (M*N*K), b is a vector with length K. I need to calculate
, which is a 2 dimensional matrix (M*N). But i don't want to write it by a explicit summation because it is too slow. Is there any way to calculate it efficiently through matlab? Please attach the code if convenience.
  3 Comments
Hancheng Zhu
Hancheng Zhu on 20 Nov 2023
Your code is very good and concise, bro. But it spends more time than the latter answer below. Thanks for your kindly help.

Sign in to comment.

Accepted Answer

Bruno Luong
Bruno Luong on 17 Nov 2023
Edited: Bruno Luong on 17 Nov 2023
This uses matrix multiplication so it should be well optimized and fast.
[m,n] = size(A, [1 2]);
C = reshape(A,m*n, []) * B(:);
C = reshape(C, m, n);

More Answers (1)

Matt J
Matt J on 17 Nov 2023
C=tensorprod(A,b,3,1)

Products


Release

R2023a

Community Treasure Hunt

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

Start Hunting!