5. Encode层
(encoder): BertEncoder(<br> (layer): ModuleList(<br> (0-11): 12 x BertLayer( # 共12层BertLayer<br> (attention): BertAttention(<br> (self): BertSelfAttention(<br> (query): Linear(in_features=768, out_features=768, bias=True)<br> (key): Linear(in_features=768, out_features=768, bias=True)<br> (value): Linear(in_features=768, out_features=768, bias=True)<br> (dropout): Dropout(p=0.1, inplace=False)<br> )<br> (output): BertSelfOutput(<br> (dense): Linear(in_features=768, out_features=768, bias=True)<br> (LayerNorm): LayerNorm((768,), eps=1e-12, elementwise_affine=True)<br> (dropout): Dropout(p=0.1, inplace=False)<br> )<br> )<br> (intermediate): BertIntermediate(<br> (dense): Linear(in_features=768, out_features=3072, bias=True)<br> (intermediate_act_fn): GELUActivation()<br> )<br> (output): BertOutput(<br> (dense): Linear(in_features=3072, out_features=768, bias=True)<br> (LayerNorm): LayerNorm((768,), eps=1e-12, elementwise_affine=True)<br> (dropout): Dropout(p=0.1, inplace=False)<br> )<br> )<br> )<br>)
Encoder层:多层Transformer编码器<br>BERT的Encoder由12个<b>BertLayer</b>堆叠而成,每个BertLayer包含三个核心部分:<br><br>Self-Attention层:捕获序列中token之间的关系。<br>Intermediate层:前馈全连接层(Feed-Forward Network)。<br>Output层:对结果进行层归一化和残差连接。