blob: 5cb86d5dc61d8b78b83cd67f4ce4f73fd7d40fe7 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
function index = mk_multi_index(n, dims, vals)
% MK_MULTI_INDEX Compute the indices of the submatrix where dims(i)=vals(i).
% index = mk_multi_index(n, dims, vals)
%
% Example:
% index = mk_multi_index(3, [1 3], [3 2])
% gives index = {3, ':', 2}, which will select out dim 1 = 3 and dim 3 = 2
% So if A(:,:,1)=[1 2;3 4; 5 6]; A(:,:,2)=[7 8; 9 10; 11 12]
% then A(index{:}) = [11 12]:
if n==0
index = { 1 };
return;
end
index = cell(1,n);
for i=1:n
index{i} = ':';
end
for i=1:length(dims)
index{dims(i)} = vals(i);
end
|
