diff --git a/beginner_source/transformer_tutorial.py b/beginner_source/transformer_tutorial.py index 8ee29ce7b4e..0ba9711ed67 100644 --- a/beginner_source/transformer_tutorial.py +++ b/beginner_source/transformer_tutorial.py @@ -86,7 +86,7 @@ def forward(self, src): src = self.pos_encoder(src) output = self.transformer_encoder(src, self.src_mask) output = self.decoder(output) - return F.log_softmax(output, dim=-1) + return output ######################################################################