How to efficiently calculate a weighted sum of 3 dimensional matrix through a vector in matlab
5 views (last 30 days)
Show older comments
A is a 3 dimensional matrix (M*N*K), b is a vector with length K. I need to calculate
![](https://www.mathworks.com/matlabcentral/answers/uploaded_files/1542397/image.png)
3 Comments
Matt J
on 20 Nov 2023
@Hancheng Zhu If that means you've settled on one of the answers below, please Accept-click one of them.
Accepted Answer
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);
0 Comments
More Answers (1)
See Also
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!