src/transformers/models/upernet/convert_swin_upernet_to_pytorch.py/correct_unfold_norm_order
def correct_unfold_norm_order(x):
    in_channel = x.shape[0]
    x = x.reshape(4, in_channel // 4)
    x = x[[0, 2, 1, 3], :].transpose(0, 1).reshape(in_channel)
    return x