MATLAB: Defining formulas for diagonal and offdiagonal elements of a matrix

Question

I have a vector M. I want to create a matrix s.t.

A(j,j) = M(j)*(1 + (M(j)-1)*roh_w)

A(j,k) = M(j)*M(k)*roh_b, whenever j!=k.

I can do

A = diag(M.*(1+(M-1)*roh_w) – M.^2*roh_b);

A = A + M.'*M*roh_b;

i.e., I first create a diagonal matrix s.t.

A(j,j) = M(j)*(1 + (M(j)-1)*roh_w) – M(j)*M(k)*roh_b,

I then add a matrix whose effect is canceled out by the second term right above for diagonal elements. Is there a prettier way of doing this?

Quintessentially, is there a one-liner code s.t. you give formula for diagonal elements and nondiagonal elements, e.g., A = FUNCTION(formula for diagonal elements, formula for nondiagonal elements)?

I'd appreciate any and all help. Thank you

Best, John

Answer 1

   A = eye(length(M))*f1+(1-eye(length(M))*f2;

where f1 is the formula for the diagonal elements and f2 that for the off-diagonal elements.