MATLAB中的元素矩阵复制

Question

我有一个3维矩阵。我想在第三维中将大小为8x2x9的矩阵复制到指定的次数，由向量说[3, 2, 1, 1, 5, 4, 2, 2, 1]给出，以便得到的矩阵的大小为8x2x21。是否有任何内置的MATLAB函数（我正在运行的版本为2014a）来执行类似于较新的repelem函数的矩阵？

我需要的一个简单例子：

% Input:
A(:,:,1) = [1 2; 1 2];
A(:,:,2) = [2 3; 2 3];

% Function call:
A = callingfunction(A, 1, 1, [1 2]);

% Output:
A(:,:,1) = [1 2; 1 2];
A(:,:,2) = [2 3; 2 3];
A(:,:,3) = [2 3; 2 3];

Answer 1

对于R2015a和更新......

根据repelem的文档（在R2015a版本中首次介绍），它也可以在矩阵上运行。我相信以下代码应该完成你想要的（我无法测试它，因为我有一个旧版本）：

newMat = repelem(mat, 1, 1, [3 2 1 1 5 4 2 2 1]);

对于R2015a之前的版本......

您可以使用this question中的一种方法将索引复制到第三维，然后使用该索引简单地索引矩阵。例如（改编Divakar's solution）：

vals = 1:size(mat, 3);
clens = cumsum([3 2 1 1 5 4 2 2 1]);
index = zeros(1, clens(end));
index([1 clens(1:end-1)+1]) = diff([0 vals]);
newMat = mat(:, :, cumsum(index));

然后，您可以将其概括为一个函数，以便在多个维度上运行，例如repelem：

function A = my_repelem(A, varargin)

  index = cell(1, nargin-1);
  for iDim = 1:nargin-1
    lens = varargin{iDim};
    if isscalar(lens)
      if (lens == 1)
        index{iDim} = ':';
        continue
      else
        lens = repmat(lens, 1, size(A, iDim));
      end
    end
    vals = 1:size(A, iDim);
    clens = cumsum(lens);
    index{iDim} = zeros(1, clens(end));
    index{iDim}([1 clens(1:end-1)+1]) = diff([0 vals]);
    index{iDim} = cumsum(index{iDim});
  end
  A = A(index{:});

end

对于您的样本数据，您可以这样使用它：

>> A(:,:,1) = [1 2; 1 2];
>> A(:,:,2) = [2 3; 2 3];
>> A = my_repelem(A, 1, 1, [1 2])

A(:,:,1) =

     1     2
     1     2

A(:,:,2) =

     2     3
     2     3

A(:,:,3) =

     2     3
     2     3