https://tanvirislam3113.blogspot.com/2023/11/diagonal-matrix-of-matrix-with-numpy.html