### 回答1:
以下是一个简单的Transformer模型的Python代码示例:
```python
import torch
import torch.nn as nn
import torch.nn.functional as F
class TransformerEncoderLayer(nn.Module):
def __init__(self, d_model, nhead, dim_feedforward=2048, dropout=0.1):
super().__init__()
self.self_attn = nn.MultiheadAttention(d_model, nhead)
self.norm1 = nn.LayerNorm(d_model)
self.dropout1 = nn.Dropout(dropout)
self.linear1 = nn.Linear(d_model, dim_feedforward)
self.activation = nn.ReLU()
self.linear2 = nn.Linear(dim_feedforward, d_model)
self.norm2 = nn.LayerNorm(d_model)
self.dropout2 = nn.Dropout(dropout)
def forward(self, src):
src2 = self.self_attn(src, src, src)[0]
src = src + self.dropout1(src2)
src = self.norm1(src)
src2 = self.linear2(self.dropout(self.activation(self.linear1(src))))
src = src + self.dropout2(src2)
src = self.norm2(src)
return src
class TransformerEncoder(nn.Module):
def __init__(self, num_layers, d_model, nhead, dim_feedforward=2048, dropout=0.1):
super().__init__()
self.layers = nn.ModuleList([TransformerEncoderLayer(d_model, nhead, dim_feedforward, dropout) for _ in range(num_layers)])
self.norm = nn.LayerNorm(d_model)
def forward(self, src):
for layer in self.layers:
src = layer(src)
src = self.norm(src)
return src
class Transformer(nn.Module):
def __init__(self, num_layers, d_model, nhead, dim_feedforward=2048, dropout=0.1):
super().__init__()
self.encoder = TransformerEncoder(num_layers, d_model, nhead, dim_feedforward, dropout)
self.decoder = nn.Linear(d_model, 1)
def forward(self, src):
src = self.encoder(src)
src = self.decoder(src)
return src
```
这里定义了一个简单的Transformer模型,包含了TransformerEncoderLayer、TransformerEncoder和Transformer三个类。其中,TransformerEncoderLayer是Transformer编码器中的一个子层,用于处理输入序列信息;TransformerEncoder是多个TransformerEncoderLayer的堆叠;Transformer则是包含一个TransformerEncoder和一个输出层的完整的Transformer模型。
代码中使用了PyTorch框架,包含了一些常用的神经网络模块,如Linear、LayerNorm、Dropout等。TransformerEncoderLayer中的self_attn是多头注意力机制,用于计算输入序列的自注意力分数和上下文向量;norm1和norm2是Layer Normalization层,用于规范化输入序列和子层输出;dropout1和dropout2是Dropout层,用于防止模型过拟合;linear1和linear2是全连接层,用于对序列信息进行变换。
在这个简单的Transformer模型中,我们只使用了一个TransformerEncoder,而没有使用TransformerDecoder,因此它只能进行单向的序列转换。如果需要进行双向的序列转换,需要使用TransformerDecoder和Masked Multi-Head Attention等模块。
### 回答2:
Transformer是一种用于自然语言处理任务的神经网络模型,其主要应用于机器翻译和文本生成任务。下面是一个简单的Transformer模型的代码实现。
代码实现使用Python编写,主要依赖于深度学习框架PyTorch。首先,需要导入相应的库和包:
```python
import torch
import torch.nn as nn
import torch.nn.functional as F
```
接下来,定义Transformer的一个关键组件——多头自注意力机制(Multi-Head Attention)。该组件通过对输入进行线性变换、注意力计算和线性变换的操作来增强特征表达能力。
```python
class MultiHeadAttention(nn.Module):
def __init__(self, d_model, num_heads):
super(MultiHeadAttention, self).__init__()
self.num_heads = num_heads
self.d_model = d_model
self.d_head = d_model // num_heads
self.query_linear = nn.Linear(d_model, d_model)
self.key_linear = nn.Linear(d_model, d_model)
self.value_linear = nn.Linear(d_model, d_model)
self.final_linear = nn.Linear(d_model, d_model)
def forward(self, queries, keys, values, mask=None):
Q = self.query_linear(queries)
K = self.key_linear(keys)
V = self.value_linear(values)
Q = self._split_heads(Q)
K = self._split_heads(K)
V = self._split_heads(V)
scores = torch.matmul(Q, K.transpose(-1, -2)) / torch.sqrt(torch.tensor(self.d_head).float())
if mask is not None:
mask = mask.unsqueeze(1)
scores = scores.masked_fill(mask == 0, -1e9)
scores = F.softmax(scores, dim=-1)
attention = torch.matmul(scores, V)
attention = self._combine_heads(attention)
attention = self.final_linear(attention)
return attention
def _split_heads(self, x):
batch_size, seq_len, d_model = x.size()
x = x.view(batch_size, seq_len, self.num_heads, self.d_head)
x = x.permute(0, 2, 1, 3).contiguous()
return x.view(batch_size * self.num_heads, seq_len, self.d_head)
def _combine_heads(self, x):
batch_size, seq_len, d_model = x.size()
x = x.view(batch_size // self.num_heads, self.num_heads, seq_len, d_model)
x = x.permute(0, 2, 1, 3).contiguous()
return x.view(batch_size // self.num_heads, seq_len, self.num_heads * d_model)
```
接下来,定义Transformer的另一个关键组件——位置编码(Positional Encoding)。位置编码用于为输入序列中的每个位置添加一组固定的向量,以表示其位置信息。
```python
class PositionalEncoding(nn.Module):
def __init__(self, d_model, max_len=5000):
super(PositionalEncoding, self).__init__()
self.d_model = d_model
pe = torch.zeros(max_len, d_model)
position = torch.arange(0, max_len, dtype=torch.float).unsqueeze(1)
div_term = torch.exp(torch.arange(0, d_model, 2).float() * (-math.log(10000.0) / d_model))
pe[:, 0::2] = torch.sin(position * div_term)
pe[:, 1::2] = torch.cos(position * div_term)
pe = pe.unsqueeze(0)
self.register_buffer('pe', pe)
def forward(self, x):
x = x + self.pe[:, :x.size(1)]
return x
```
最后,定义完整的Transformer模型。
```python
class Transformer(nn.Module):
def __init__(self, d_model, num_heads, num_encoder_layers, num_decoder_layers, feed_forward_dim, src_vocab_size, tgt_vocab_size, max_len):
super(Transformer, self).__init__()
self.encoder = nn.Embedding(src_vocab_size, d_model)
self.src_positional_encoding = PositionalEncoding(d_model, max_len)
self.decoder = nn.Embedding(tgt_vocab_size, d_model)
self.tgt_positional_encoding = PositionalEncoding(d_model, max_len)
self.encoder_layers = nn.ModuleList([nn.TransformerEncoderLayer(d_model, num_heads, feed_forward_dim) for _ in range(num_encoder_layers)])
self.decoder_layers = nn.ModuleList([nn.TransformerDecoderLayer(d_model, num_heads, feed_forward_dim) for _ in range(num_decoder_layers)])
self.fc = nn.Linear(d_model, tgt_vocab_size)
def forward(self, src, tgt, src_mask=None, tgt_mask=None):
src = self.encoder(src) * math.sqrt(self.d_model)
src = self.src_positional_encoding(src)
tgt = self.decoder(tgt) * math.sqrt(self.d_model)
tgt = self.tgt_positional_encoding(tgt)
for layer in self.encoder_layers:
src = layer(src, src_mask)
for layer in self.decoder_layers:
tgt = layer(tgt, src, tgt_mask)
output = self.fc(tgt)
return output
```
以上是一个简单的Transformer模型的代码实现。通过定义多头自注意力机制、位置编码和完整的Transformer模型,可以用于自然语言处理任务的实现。