Matlab：矩阵中每一行的Argmax和点积

我有2个矩阵=X in R^(n*m) and W in R^(k*m) where k<<n. Let x_i是 X 的第 i 行并且w_j是 W 的第 j 行。 我需要找到对于每个 x_i 最大化的 j 是什么<w_j,x_i>

我看不到迭代 X 中所有行的方法，但有没有一种方法可以找到最大点积，而无需每次迭代所有 W ？

一个简单的实现是：

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

点积本质上是矩阵乘法：

[~, Y] = max(W*X');