Vectorizing combined matrix and element-wise multiplication in MATLAB

Question

I am looking for a way to vectorize a combined matrix multiplication and element-wise addition.

Let's say I have a matrix function M_{ij}(x), and a vector function v_j(x), where {i,j} are matrix indices, and x is a position variable. I want to perform an element-wise matrix multiplication and find u(x) = M(x).v(x). A straightforward example is:

imax = 2; jmax = 3; xmax = 10;

M=rand(imax,jmax,xmax);
v=rand(jmax,xmax);
u=zeros(imax,xmax);

for i=1:imax
    for j=1:jmax
        u(i,:) = u(i,:) + squeeze(M(i,j,:))'.*v(j,:);
    end
end

Is there a vectorized way to speed up this operation? In my problem we will assume that imax,jmax are <5, and that xmax is large.

chaohuang · Accepted Answer · 2012-08-02 17:16:25Z

1

Try

 u=squeeze(sum(bsxfun(@times,permute(M, [2 3 1]), v)))';

answered Aug 2, 2012 at 17:16

chaohuang

4,1254 gold badges33 silver badges37 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Vectorizing combined matrix and element-wise multiplication in MATLAB

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related