wehub-resource-sync
17 lines
443 B
Matlab
Executable File
17 lines
443 B
Matlab
Executable File
function B = sub(A, varargin)
|
|
% sub(A,i,j,k) = A(i;j;k)
|
|
% Written by Mo Chen (sth4nth@gmail.com).
|
|
assert(ndims(A)==numel(varargin));
|
|
sz = cellfun(@numel,varargin);
|
|
IDX = cell(1,ndims(A));
|
|
for i = 1:ndims(A)
|
|
idx = varargin{i};
|
|
shape = ones(1,ndims(A));
|
|
shape(i) = sz(i);
|
|
idx = reshape(idx,shape);
|
|
shape = sz;
|
|
shape(i) = 1;
|
|
idx = repmat(idx,shape);
|
|
IDX{i} = idx(:);
|
|
end
|
|
B = reshape(A(sub2ind(size(A),IDX{:})),sz); |