fisher_block_snn.m

来自「神经网络的工具箱, 神经网络的工具箱,」· M 代码 · 共 39 行

function F = fisher_block_snn(X, net, data)
%FISHER_BLOCK_SNN Block estimate of fisher matrix.
%
% Syntax 
%
%   F = fisher_block_snn(net, data);
%   F = fisher_block_snn(X, net, data);
%

if (nargin == 2)
   data = net;
   net = X;
elseif (nargin == 3)
   net = setx_snn(net, X);
else
   error('FISHER_BLOCK_SNN: Incorrect number of input arguments');
end

F = fisherff_snn(net, data);

L = net.numLayers;
ind_b = zeros(L,1);
ind_e = zeros(L,1);
ind_b(1) = 1;
ind_e(1) = prod(size(net.weights{1})+[0 1]);
for l = 2:L
    ind_b(l) = ind_e(l-1) + 1;
    ind_e(l) = ind_e(l-1) + prod(size(net.weights{l})+[0 1]);
end

F([ind_b(1):ind_e(1)], [ind_b(2):ind_e(L)]) = 0;
for l = 2:(L-1)
    F([ind_b(l):ind_e(l)], [ind_b(1):ind_e(l-1)]) = 0;
    F([ind_b(l):ind_e(l)], [ind_b(l+1):ind_e(L)]) = 0;
end
F([ind_b(L):ind_e(L)], [ind_b(1):ind_e(L-1)]) = 0;