#A trick when you want to flatten a matrix X of shape (a,b,c,d)... #...to a matrix X_flatten of shape (b ∗ c ∗ d, a) is to use: X_flatten = X.reshape(X.shape[0], -1).T # X.T is the transpose of X