def METHOD_NAME(params):
    reordered_weights = []
    for (name, weight) in params: # param = ("name", weight)
        weight = weight.hsplit(3)
        reordered_weights.append((name, torch.hstack((weight[1], weight[0], weight[2])))) # reorder
    return reordered_weights