Matlab: Argmax e produto escalar para cada linha em uma matriz

Eu tenho 2 matrizes =X in R^(n*m) eW in R^(k*m) Ondek<<n. Deixeix_i ser a i-ésima linha de X ew_j seja a j-ésima linha de W. Preciso encontrar, para cada x_i qual é o j que maximiza<w_j,x_i>

Não consigo ver uma maneira de iterar em todas as linhas em X, mas existe uma maneira de encontrar o produto com pontos máximos sem iterar todas as vezes em todo o W?

Uma implementação ingênua seria:

n = 100;
m = 50;
k = 10;
X = rand(n,m);
W = rand(k,m);
Y = zeros(n, 1);

for i = 1 : n
  max_ind = 1;
  max_val = dot(W(1,:), X(i,:));
  for j = 2 : k
       cur_val = dot(W(j,:),X(i,:));

       if cur_val > max_val
          max_val = cur_val;
          max_ind = j;
       end

   end

   Y(i,:) = max_ind;
end

questionAnswers(2)

yourAnswerToTheQuestion