From dda235279ab8ba716ff2ea3fc31dfc69a7e871b0 Mon Sep 17 00:00:00 2001 From: Amit Aharoni Date: Mon, 27 Oct 2025 21:47:11 +0100 Subject: [PATCH] Fixed incorrect description of what each dimension represents --- beginner_source/basics/buildmodel_tutorial.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/beginner_source/basics/buildmodel_tutorial.py b/beginner_source/basics/buildmodel_tutorial.py index 1806e80feb5..e70aa8e8218 100644 --- a/beginner_source/basics/buildmodel_tutorial.py +++ b/beginner_source/basics/buildmodel_tutorial.py @@ -75,7 +75,7 @@ def forward(self, x): # along with some `background operations `_. # Do not call ``model.forward()`` directly! # -# Calling the model on the input returns a 2-dimensional tensor with dim=0 corresponding to each output of 10 raw predicted values for each class, and dim=1 corresponding to the individual values of each output. +# Calling the model on the input returns a 2-dimensional tensor with dim=0 corresponding to the batch dimension (one entry per example in the input) and dim=1 corresponding to the 10 raw predicted class scores (logits) for each example. # We get the prediction probabilities by passing it through an instance of the ``nn.Softmax`` module. X = torch.rand(1, 28, 28, device=device)