wolffd@0: function r = regularizeMKLDiag(W, X, gradient)
wolffd@0: %
wolffd@0: % r = regularizeMKL(W, X, gradient)
wolffd@0: %
wolffd@0: %
wolffd@0: 
wolffd@0:     [d,n,m] = size(X);
wolffd@0: 
wolffd@0:     if gradient
wolffd@0:         r = zeros(d,m);
wolffd@0:         for i = 1:m
wolffd@0:             r(:,i) = diag(X(:,:,i));
wolffd@0:         end
wolffd@0:     else
wolffd@0:         r = 0;
wolffd@0:         for i = 1:m
wolffd@0:             r = r + W(:,i)' * diag(X(:,:,i));
wolffd@0:         end
wolffd@0:     end
wolffd@0: end