sumv.m

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

function T2 = sumv(T1, sum_over)
%
% Like the built in sum, but will sum over several dimensions and then squeeze the result.

T2 = T1;
for i=1:length(sum_over)
  if sum_over(i) <= ndims(T2) % prevent summing over non-existent dimensions
    T2=sum(T2, sum_over(i));
  end
end
T2 = squeeze(T2);