开始使用免费开始使用

PyTorch Transformer

现在您已经熟悉了 Transformer 架构的各个组件,是时候来定义一个模型了!torch.nn 模块(已为您以 nn 导入)提供了一种非常简洁的方式,只需几行代码即可完成。

本练习是课程的一部分

使用 PyTorch 的 Transformer 模型

查看课程

练习说明

  • 定义一个 Transformer:使用 8 个注意力头,6 个编码器层和 6 个解码器层,输入序列嵌入向量长度为 1536
  • 打印模型对象以查看模型结构。

交互式实操练习

通过完成这段示例代码来试试这个练习。

# Define the transformer model
model = ____

# Print the model object
print(____)
编辑并运行代码