I\'m trying to vectorize this function, but nothing I try works. It is supposed to make an n x n matrix with 2\'s along its diagonal and -1 on the first upper and lower diag