Return X.permute 0 2 1 3