Pytorch 这就是nn.Transformer的工作原理吗？

如果我想将一个图像转换为另一个图像， 然后

---

这是使用nn.Transformer的正确方法吗？

不，这不是Transformer模块的功能。Transformer主要用于对大型文本体上的NLP通用模型进行预培训。如果您想了解更多信息，我强烈建议您阅读介绍该体系结构的文章。如果你听说过这样的模型，这些就是变压器的例子

当你问到如何“将一个图像转换成另一个图像”时，你并不完全清楚你想要完成什么。我想也许你在寻找这样的东西

无论如何，要重新回答您的问题：不，这不是您使用nn.Transformer的方式。你应该尝试澄清你试图通过“将一张图片转换成另一张图片”来完成什么，并将描述作为一个单独的问题发布

transformer_model = nn.Transformer(img_size, n_heads)
transformer_model(source_image, target_image)