对一句话打乱token (input_ids)

def shuffle_token(vector):
    # 获取词向量长度
    second_dim_length = vector.shape[-1]

    # 生成一个与词向量长度相同的随机排列
    permutation = list(range(second_dim_length))
    random.shuffle(permutation)

    # 根据随机排列重新排列词向量
    shuffled_vector = vector[:, permutation]

    return shuffled_vector

你可能感兴趣的:(python,深度学习,人工智能)