https://mingredbestcheapprice.blogspot.com/2024/01/3d-numpy-array-into-block-diagonal.html