setdiag.m

The tca package is a Matlab program that implements the tree-dependent component analysis (TCA) alg

function M = setdiag(M, v)
% SETDIAG Set the diagonal of a matrix to a specified scalar/vector.
% M = set_diag(M, v)

n = length(M);
if length(v)==1
  v = repmat(v, 1, n);
end
if issparse(M)
  M = triu(M,1) + tril(M,-1) + sparse(diag(v));
else 
  M = triu(M,1) + tril(M,-1) + diag(v);
end