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