PyTorch Transformer
现在您已经熟悉了 Transformer 架构的各个组件,是时候来定义一个模型了!torch.nn 模块(已为您以 nn 导入)提供了一种非常简洁的方式,只需几行代码即可完成。
本练习是课程的一部分
使用 PyTorch 的 Transformer 模型
练习说明
- 定义一个 Transformer:使用
8个注意力头,6个编码器层和6个解码器层,输入序列嵌入向量长度为1536。 - 打印模型对象以查看模型结构。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Define the transformer model
model = ____
# Print the model object
print(____)