Mercurial > hg > camir-aes2014
annotate toolboxes/FullBNT-1.0.7/KPMtools/setdiag.m @ 0:e9a9cd732c1e tip
first hg version after svn
author | wolffd |
---|---|
date | Tue, 10 Feb 2015 15:05:51 +0000 |
parents | |
children |
rev | line source |
---|---|
wolffd@0 | 1 function M = setdiag(M, v) |
wolffd@0 | 2 % SETDIAG Set the diagonal of a matrix to a specified scalar/vector. |
wolffd@0 | 3 % M = set_diag(M, v) |
wolffd@0 | 4 |
wolffd@0 | 5 n = length(M); |
wolffd@0 | 6 if length(v)==1 |
wolffd@0 | 7 v = repmat(v, 1, n); |
wolffd@0 | 8 end |
wolffd@0 | 9 |
wolffd@0 | 10 % e.g., for 3x3 matrix, elements are numbered |
wolffd@0 | 11 % 1 4 7 |
wolffd@0 | 12 % 2 5 8 |
wolffd@0 | 13 % 3 6 9 |
wolffd@0 | 14 % so diagnoal = [1 5 9] |
wolffd@0 | 15 |
wolffd@0 | 16 |
wolffd@0 | 17 J = 1:n+1:n^2; |
wolffd@0 | 18 M(J) = v; |
wolffd@0 | 19 |
wolffd@0 | 20 %M = triu(M,1) + tril(M,-1) + diag(v); |
wolffd@0 | 21 |