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

    return x