blob: a8e26c254ebac7ef01b159c86ec8a1fce001767f (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
function strides = compute_strides(bnet)
% COMPUTE_STRIDES For each CPT and each variable in that CPT,
% returns the stride of that variable. So in future, we can
% quickly extract a slice of the CPT.
%
% The return value is a 2d array, where strides(i,j) contains the
% stride of the jth variable in the ith CPT. Cell arrays would
% have saved space but they are slower.
%
num_cpts = size(bnet.CPD, 2);
max_cpt_dim = 1 + max(sum(bnet.dag));
strides = zeros(num_cpts, max_cpt_dim);
for i = 1:num_cpts
c = CPT(bnet, i);
siz = size(CPT(bnet, i));
% Deal with the special case of a 1-d array separately
if siz(2) == 1
dim = 1;
else
dim = size(siz, 2);
end
strides(i, 1:dim ) = [1 cumprod(siz(1:dim-1))];
end
|
