wolffd@0: function r = regularizeTraceDiag(W, X, gradient)
wolffd@0: %
wolffd@0: % r = regularizeTraceDiag(W, X, gradient)
wolffd@0: %
wolffd@0: %
wolffd@0:     d = length(W);
wolffd@0: 
wolffd@0:     if gradient
wolffd@0:         r = ones(d,1) / d;
wolffd@0:     else
wolffd@0:         r = sum(W) / d;
wolffd@0:     end
wolffd@0: end