return X.reshape(-1, X.shape[2], X.shape[3])