matlab - 在 MATLAB 中替换 repmat

Question

我有一个函数可以多次执行以下循环:

for cluster=1:max(bins), % bins is a list in the same format as kmeans() IDX output
    select=bins==cluster; % find group of values
    means(select,:)=repmat_fast_spec(meanOneIn(x(select,:)),sum(select),1); 
    % (*, above) for each point, write the mean of all points in x that 
    % share its label in bins to the equivalent row of means
    delta_x(select,:)=x(select,:)-(means(select,:)); 
    %subtract out the mean from each point
end

注意到 repmat_fast_spec 和 meanOneIn 是 repmat() 和 mean() 的精简版本，分别地，我想知道是否有一种方法可以在标记为 (*) 的行中进行分配，从而完全避免 repmat。

关于如何从这件事中榨取性能的任何其他想法也将受到欢迎。

Answer 1

这里有一个可能的改进来避免 REPMAT:

x = rand(20,4);
bins = randi(3,[20 1]);

d = zeros(size(x));
for i=1:max(bins)
    idx = (bins==i);
    d(idx,:) = bsxfun(@minus, x(idx,:), mean(x(idx,:)));
end

另一种可能性:

x = rand(20,4);
bins = randi(3,[20 1]);

m = zeros(max(bins),size(x,2));
for i=1:max(bins)
    m(i,:) = mean( x(bins==i,:) );
end
dd = x - m(bins,:);