marg_table_ndxb.m

来自「麻省理工学院的人工智能工具箱,很珍贵,希望对大家有用!」· M 代码 · 共 27 行

function smallT = marg_table_ndxB(bigT, maximize, ndx)
% MARG_TABLE_NDXB Marg a dtable onto a smaller domain.
% smallT = marg_table_ndxB(bigT, maximize, ndx)

N = prod(size(bigT));
S = size(ndx, 1);
smallT = zeros(1,S);
if S==1
   if maximize
      smallT = max(bigT);
   else
      smallT = sum(bigT);
   end
   return
end

ndx = double(ndx)+1;
if maximize
   for i = 1:N
      smallT(ndx(i)) = max(smallT(ndx(i)) + bigT(i));
   end
else
   for i = 1:N
      smallT(ndx(i)) = smallT(ndx(i)) + bigT(i);
   end
end