帮我写一个bert-bilstm-crf-ner模型用于中文命名实体识别

好的，下面是bert-bilstm-crf-ner模型的代码实现（Python）：

import torch
import torch.nn as nn
from transformers import BertModel

class BertBilstmCrfNer(nn.Module):
    def __init__(self, num_tags, hidden_dim=768, bert_model_name='bert-base-chinese'):
        super(BertBilstmCrfNer, self).__init__()
        self.bert = BertModel.from_pretrained(bert_model_name)
        self.dropout = nn.Dropout(0.5)
        self.bilstm = nn.LSTM(hidden_dim, hidden_dim // 2, num_layers=2, bidirectional=True, batch_first=True)
        self.fc = nn.Linear(hidden_dim, num_tags)
        self.crf = nn.CRF(num_tags, batch_first=True)

    def forward(self, input_ids, attention_mask, labels=None):
        bert_output = self.bert(input_ids=input_ids, attention_mask=attention_mask)[0]
        bert_output = self.dropout(bert_output)
        lstm_output, _ = self.bilstm(bert_output)
        logits = self.fc(lstm_output)
        if labels is not None:
            loss = -self.crf(logits, labels, mask=attention_mask.byte(), reduction='mean')
            return loss
        else:
            pred_labels = self.crf.decode(logits, mask=attention_mask.byte())
            return pred_labels

该模型使用了BERT作为预训练模型，使用了BiLSTM作为特征提取器，并使用CRF作为序列标注器。在模型中，我们首先使用BERT模型对输入进行编码，然后通过BiLSTM提取特征，最后通过全连接层将提取的特征映射到标签空间。在训练时，我们使用CRF计算损失，并通过反向传播更新模型参数。在推理时，我们使用CRF进行解码，得到最终的标签序列。

模型的输入是一个batch的文本序列，每个文本序列已经使用BERT进行编码，因此输入是一个二维张量，形状为(batch_size, sequence_length, hidden_dim)。另外，我们还需要提供一个attention_mask张量，用于指示输入中的实际文本序列部分。标签序列是一个二维张量，形状为(batch_size, sequence_length)，每个元素代表一个标签。在训练时，我们需要提供标签序列，用于计算损失；在推理时，我们不需要提供标签序列，而是通过CRF进行解码，得到最终的标签序列。