samer@4: function A=setdiag(A,x)
samer@4: % setdiag - set elements on diagonal to something
samer@4: %
samer@4: % setdiag :: [[N,M]], real -> [[N,M]].
samer@4: % setdiag :: [[N,N]], [[N]] -> [[N,N]].
samer@4: 
samer@4: dom=size(A);
samer@4: i=1:min(dom);
samer@4: A(sub2ind(dom,i,i))=x;