帮我利用Bert和pytorch写一个文本的Encoder的代码

好的，以下是使用Bert和pytorch编写文本Encoder的代码：

import torch
from transformers import BertTokenizer, BertModel

tokenizer = BertTokenizer.from_pretrained('bert-base-chinese')
model = BertModel.from_pretrained('bert-base-chinese')

def encode_text(text):
    input_ids = torch.tensor([tokenizer.encode(text, add_special_tokens=True)])
    with torch.no_grad():
        outputs = model(input_ids)
        encoded_text = outputs[0][:, 0, :]
    return encoded_text

这个函数可以将输入的文本编码为一个向量，可以用于文本分类、文本相似度计算等任务。