i want to get a diagonal matrix from this: [[1 2 3] [4 5 6] [7 8 9]]. then it occurred to me to do the following:
[[1 2 3] [4 5 6] [7 8 9]]
[[0 0 0] [4 0 0] [7 8 0]] + [[0