📌  相关文章
📜  pytorch 中的变压器 - 任何代码示例

📅  最后修改于: 2022-03-11 14:59:57.841000             🧑  作者: Mango

代码示例1
>>> transformer_model = nn.Transformer(nhead=16, num_encoder_layers=12)
>>> src = torch.rand((10, 32, 512))
>>> tgt = torch.rand((20, 32, 512))
>>> out = transformer_model(src, tgt)