How to apply 2D mask on a specific side of 3D array?

Question

Suppose we have regular 3D array of shape (n,n,n) padded with Trues. I would like to create an array of diagonal principal minors. They should be m