Fill an array with matrices along thr k-th diagonal

Is there an elegant way or function to fill a square 2D-array along its k-th diagonal with sub-matrices?