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