Нет, это не то, что делает модуль Transformer. Трансформатор в основном используется для предварительной подготовки моделей общего пользования для НЛП на больших объемах текста. Если вам интересно узнать больше, я настоятельно рекомендую вам прочитать статью, в которой представлена архитектура «Внимание - это все, что вам нужно» . Если вы слышали о таких моделях, как BERT или GPT-2 , это примеры трансформаторов.
Не совсем ясно, чего вы пытаетесь достичь, когда спрашиваете, как «преобразоватьизображение в другое изображение. "Я думаю, может быть, вы ищете что-то такое? https://junyanz.github.io/CycleGAN/
В любом случае, чтобы ответить на ваш вопрос: нет, вы не так используете nn.Transformer. Вы должны попытаться уточнить, что вы пытаетесь сделать с помощью «преобразования одной картинки в другую», и опубликовать это описание как отдельный вопрос.