def swap_list_order(alist):
    """
    Args:
        alist: shape=(B, num_level, ....)
    Returns:
        alist: shape=(num_level, B, ...)
    """
    new_order0 = len(alist[0])
    return [[alist[i][j] for i in range(len(alist))] for j in range(new_order0)]