machine-learning - PyTorch 的 nn.Transformer 中的 src 和 tgt 有什么区别？

Question

从文档中可以看出，要创建这样的转换器模型:

transformer_model = nn.Transformer(nhead=16, num_encoder_layers=12)
src = torch.rand((10, 32, 512))
tgt = torch.rand((20, 32, 512)) # What is tgt??
out = transformer_model(src, tgt) 

tgt 意味着什么？ tgt 应该和 src 一样吗？

Answer 1

Transformer 结构由两个组件组成，编码器和解码器。 src是编码器的输入，tgt是解码器的输入。

例如执行将英语句子翻译成法语的机器翻译任务，src 是英语序列 id，tgt 是法语序列 id。