def get_embeddings(pretrain_model, batch_tokens):
    """
       Get avg pooling of the embedding of the input sequence data
       :param batch_tokens: list[tokens]
       :return: tensor: [n,1280]
    """
    results = pretrain_model(batch_tokens, repr_layers=[33], return_contacts=True)
    token_representations = results["representations"][33]  # [num,maxlen,embed=1280]
    pool_embedding = token_representations.mean(1)
    return pool_embedding