I have matrix Q
Q.shape is (2,2,n,n). Where n is k**2 , k > 1
Q.shape
(2,2,n,n)
n
k**2
So Q is block matrix:
Q[0,0].shap